apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix rounding issue that caused 1px box shift for YOLO format. (#730)

Browse Source
This commit is contained in:
Denis
2021-04-07 02:38:48 +03:00
committed by GitHub
parent 3425fefd8d
commit 0573a39ec3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/yolo_io.py
+4 -4
View File
@@ -129,10 +129,10 @@ class YoloReader:
y_min = max(float(y_center) - float(h) / 2, 0) y_min = max(float(y_center) - float(h) / 2, 0)
y_max = min(float(y_center) + float(h) / 2, 1) y_max = min(float(y_center) + float(h) / 2, 1)
x_min = int(self.img_size[1] * x_min) x_min = round(self.img_size[1] * x_min)
x_max = int(self.img_size[1] * x_max) x_max = round(self.img_size[1] * x_max)
y_min = int(self.img_size[0] * y_min) y_min = round(self.img_size[0] * y_min)
y_max = int(self.img_size[0] * y_max) y_max = round(self.img_size[0] * y_max)
return label, x_min, y_min, x_max, y_max return label, x_min, y_min, x_max, y_max