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

Cleanup additional tensor in detection worker
All checks were successful
Build Dev PWA / Build-PWA (push) Successful in 37s
Details

Browse Source 
Signed-off-by: Justin Georgi <justin.georgi@gmail.com>
This commit is contained in:
Justin Georgi
2024-10-12 15:25:53 -07:00
parent 21e46713a7
commit 8bf74e51ea
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/assets/detect-worker.js
View File

@@ -78,7 +78,7 @@ async function localDetect(imageData) {
const output = { const output = {
detections: [] detections: []
} }
let rawBoxes = [] let rawBoxes = []
let rawScores = [] let rawScores = []
let getScores, getBox, boxCalc let getScores, getBox, boxCalc
@@ -123,6 +123,7 @@ let rawBoxes = []
tf.dispose(tBoxes) tf.dispose(tBoxes)
tf.dispose(tScores) tf.dispose(tScores)
tf.dispose(tRes) tf.dispose(tRes)
tf.dispose(resBoxes)
const valid_detections_data = classes_data.length const valid_detections_data = classes_data.length
for (let i =0; i < valid_detections_data; i++) { for (let i =0; i < valid_detections_data; i++) {
let [dLeft, dTop, dRight, dBottom] = boxes_data[i] let [dLeft, dTop, dRight, dBottom] = boxes_data[i]
@@ -131,7 +132,6 @@ let rawBoxes = []
"left": dLeft, "left": dLeft,
"bottom": dBottom, "bottom": dBottom,
"right": dRight, "right": dRight,
// "label": this.detectorLabels[classes_data[i]].name,
"label": classes_data[i], "label": classes_data[i],
"confidence": scores_data[i] * 100 "confidence": scores_data[i] * 100
}) })