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)