Skip to content

Commit bb8c68d

Browse files
Check for bounding box coordinates for Yolo
Suggested fix for cvat-ai#896
1 parent c0f1854 commit bb8c68d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

utils/open_model_zoo/yolov3/interp.py

+6
Original file line numberDiff line numberDiff line change
@@ -150,5 +150,11 @@ def parse_yolo_region(self, blob: 'np.ndarray', original_shape: list, params: di
150150
xmax = obj['xmax']
151151
ymin = obj['ymin']
152152
ymax = obj['ymax']
153+
154+
# Enforcing checks
155+
xmin = max(0,xmin)
156+
ymin = max(0,ymin)
157+
xmax = min(xmax,width)
158+
ymax = min(ymax,height)
153159

154160
results.add_box(xmin, ymin, xmax, ymax, label, frame_number)

0 commit comments

Comments
 (0)