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

Change process to use arraySync and filter resulta

Browse Source 
Signed-off-by: Justin Georgi <justin.georgi@gmail.com>
This commit is contained in:
Justin Georgi
2024-03-08 15:39:42 -07:00
parent 1cc76c5950
commit 251751c7bf
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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