Fix cocossd bounding boxes

bomanimc · Nov 22, 2019 · c660e16 · c660e16
1 parent 2dc6f5c
commit c660e16
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/ObjectDetector/CocoSsd/index.js b/src/ObjectDetector/CocoSsd/index.js
@@ -67,10 +67,10 @@ class CocoSsdBase {
             return {
                 label: prediction.class,
                 confidence: prediction.score,
-                x: prediction.bbox[0],
-                y: prediction.bbox[1],
-                w: prediction.bbox[2],
-                h: prediction.bbox[3],
+                x: prediction.bbox[0] / imgToPredict.width,
+                y: prediction.bbox[1] / imgToPredict.height,
+                w: prediction.bbox[2] / imgToPredict.width,
+                h: prediction.bbox[3] / imgToPredict.height,
             }
         })