ALVINN/ALVINN_f7
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects Releases 7 Wiki Activity

Optimize detection steps #129

Merged
jgeorgi merged 2 commits from xps-detect-optim into main 2024-03-10 04:06:51 +00:00
Conversation 0 Commits 2 Files Changed 1 +6 -4
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 251751c7bf - Show all commits

10
src/pages/detection-mixin.js
View File

@@ -29,12 +29,14 @@ export default {
console.time('post-process') console.time('post-process')
const detectAttempts = res.shape[2] const detectAttempts = res.shape[2]
const outputSize = res.shape[1] const outputSize = res.shape[1]
const rawRes = tf.transpose(res,[0,2,1]).dataSync() const rawRes = tf.transpose(res,[0,2,1]).arraySync()[0]
let rawBoxes = [] let rawBoxes = []
let rawScores = [] let rawScores = []
const filteredRes = rawRes.filter( r => r.slice(4).some( s => s > .05))
for (var i = 0; i < detectAttempts; i++) { for (var i = 0; i < filteredRes.length; i++) {
var getBox = rawRes.slice((i * outputSize),(i * outputSize) + 4) var getScores = filteredRes[i].slice(4)
var getBox = filteredRes[i].slice(0,4)
var boxCalc = [ var boxCalc = [
(getBox[0] - (getBox[2] / 2)) / modelWidth, (getBox[0] - (getBox[2] / 2)) / modelWidth,
(getBox[1] - (getBox[3] / 2)) / modelHeight, (getBox[1] - (getBox[3] / 2)) / modelHeight,
@@ -42,7 +44,7 @@ export default {
(getBox[1] + (getBox[3] / 2)) / modelHeight, (getBox[1] + (getBox[3] / 2)) / modelHeight,
] ]
rawBoxes.push(boxCalc) rawBoxes.push(boxCalc)
rawScores.push(rawRes.slice((i * outputSize) + 4,(i + 1) * outputSize)) rawScores.push(getScores)
} }
const tBoxes = tf.tensor2d(rawBoxes) const tBoxes = tf.tensor2d(rawBoxes)
let tScores = null let tScores = null