🐛 Update: Bbox Coords

Team-BoonMoSa · Mar 23, 2023 · 409d298 · 409d298
1 parent ba49c77
commit 409d298
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/saveData.py b/saveData.py
@@ -55,7 +55,7 @@ def saveData(target, brand, img):
         wr = csv.writer(f, delimiter=' ')
         for i in range(len(bbox)):
             x, y, width, height = bbox.iloc[i]
-            wr.writerow([0, x / mw, y / mh, width / mw, height / mh])
+            wr.writerow([0, x / mw + mw // 2, y / mh + mh // 2, width / mw, height / mh])
 
 if __name__ == "__main__":
     os.chdir('..')