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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/yolo_io.py
View File

@ -129,10 +129,10 @@ class YoloReader:
y_min = max(float(y_center) - float(h) / 2, 0)
y_max = min(float(y_center) + float(h) / 2, 1)
x_min = int(self.img_size[1] * x_min)
x_max = int(self.img_size[1] * x_max)
y_min = int(self.img_size[0] * y_min)
y_max = int(self.img_size[0] * y_max)
x_min = round(self.img_size[1] * x_min)
x_max = round(self.img_size[1] * x_max)
y_min = round(self.img_size[0] * y_min)
y_max = round(self.img_size[0] * y_max)
return label, x_min, y_min, x_max, y_max