diff --git a/saveData.py b/saveData.py index 4e69d4b..07ff36c 100644 --- 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('..')