Check for bounding box coordinates for Yolo

Suggested fix for cvat-ai#896
himalayanZephyr · Dec 6, 2019 · bb8c68d · bb8c68d
1 parent c0f1854
commit bb8c68d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/utils/open_model_zoo/yolov3/interp.py b/utils/open_model_zoo/yolov3/interp.py
@@ -150,5 +150,11 @@ def parse_yolo_region(self, blob: 'np.ndarray', original_shape: list, params: di
             xmax = obj['xmax']
             ymin = obj['ymin']
             ymax = obj['ymax']
+
+            # Enforcing checks
+            xmin = max(0,xmin)
+            ymin = max(0,ymin)
+            xmax = min(xmax,width)
+            ymax = min(ymax,height)
 
             results.add_box(xmin, ymin, xmax, ymax, label, frame_number)