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