Fix: Do not round bbox coordinates in mode opendatacam-darknet

opendatacam · Sep 19, 2019 · 3ff1b75 · 3ff1b75
1 parent efb8033
commit 3ff1b75
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -153,7 +153,7 @@ node-moving-things-tracker --debug --input PATH_TO_YOLO_DETECTIONS.txt
 
 # Run on opendatacam/darknet detection data
 
-Usage with opendatacam/darknet generated tracker data.
+Usage with opendatacam/darknet (https://github.com/opendatacam/darknet/pull/2) generated tracker data 
 
 ```bash
 node-moving-things-tracker --mode opendatacam-darknet --input detectionsFromDarknet.json

diff --git a/main.js b/main.js
@@ -173,7 +173,12 @@ fs.readFile(`${pathRawDetectionsInput}`, function (err, f) {
     Tracker.updateTrackedItemsWithNewFrame(detectionsForThisFrame, parseInt(frameNb, 10))
 
     if (!MODE_MOTChallenge) {
-      tracker[frameNb] = Tracker.getJSONOfTrackedItems();
+      if(MODE_DARKNET) {
+        // Do not round coordinates of bbox
+        tracker[frameNb] = Tracker.getJSONOfTrackedItems(false);
+      } else {
+        tracker[frameNb] = Tracker.getJSONOfTrackedItems();
+      }
     } else {
       MOToutput = MOToutput.concat(Tracker.getTrackedItemsInMOTFormat(frameNb));
     }

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "node-moving-things-tracker",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "Tracker by detections in javascript for node.js / browsers",
   "url": "https://github.com/tdurand/node-moving-things-tracker",
   "main": "main.js",