fix_annotate bug fix

tests/ut/python/dataset/test_bounding_box_augment.py

@@ -34,12 +34,16 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for bbox in bboxes:
-        tmp = bbox[0]
+        if bbox.size == 7:
-        bbox[0] = bbox[1]
+            tmp = bbox[0]
-        bbox[1] = bbox[2]
+            bbox[0] = bbox[1]
-        bbox[2] = bbox[3]
+            bbox[1] = bbox[2]
-        bbox[3] = bbox[4]
+            bbox[2] = bbox[3]
-        bbox[4] = tmp
+            bbox[3] = bbox[4]
+            bbox[4] = tmp
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_random_crop_and_resize_with_bbox.py

@@ -36,12 +36,16 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for bbox in bboxes:
-        tmp = bbox[0]
+        if bbox.size == 7:
-        bbox[0] = bbox[1]
+            tmp = bbox[0]
-        bbox[1] = bbox[2]
+            bbox[0] = bbox[1]
-        bbox[2] = bbox[3]
+            bbox[1] = bbox[2]
-        bbox[3] = bbox[4]
+            bbox[2] = bbox[3]
-        bbox[4] = tmp
+            bbox[3] = bbox[4]
+            bbox[4] = tmp
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_random_crop_with_bbox.py

@@ -37,12 +37,16 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for bbox in bboxes:
-        tmp = bbox[0]
+        if bbox.size == 7:
-        bbox[0] = bbox[1]
+            tmp = bbox[0]
-        bbox[1] = bbox[2]
+            bbox[0] = bbox[1]
-        bbox[2] = bbox[3]
+            bbox[1] = bbox[2]
-        bbox[3] = bbox[4]
+            bbox[2] = bbox[3]
-        bbox[4] = tmp
+            bbox[3] = bbox[4]
+            bbox[4] = tmp
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_random_horizontal_flip_with_bbox.py

@@ -34,12 +34,16 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for bbox in bboxes:
-        tmp = bbox[0]
+        if bbox.size == 7:
-        bbox[0] = bbox[1]
+            tmp = bbox[0]
-        bbox[1] = bbox[2]
+            bbox[0] = bbox[1]
-        bbox[2] = bbox[3]
+            bbox[1] = bbox[2]
-        bbox[3] = bbox[4]
+            bbox[2] = bbox[3]
-        bbox[4] = tmp
+            bbox[3] = bbox[4]
+            bbox[4] = tmp
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_random_resize_with_bbox.py

@@ -35,7 +35,11 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for (i, box) in enumerate(bboxes):
-        bboxes[i] = np.roll(box, -1)
+        if box.size == 7:
+            bboxes[i] = np.roll(box, -1)
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_random_vertical_flip_with_bbox.py

@@ -36,12 +36,16 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for bbox in bboxes:
-        tmp = bbox[0]
+        if bbox.size == 7:
-        bbox[0] = bbox[1]
+            tmp = bbox[0]
-        bbox[1] = bbox[2]
+            bbox[0] = bbox[1]
-        bbox[2] = bbox[3]
+            bbox[1] = bbox[2]
-        bbox[3] = bbox[4]
+            bbox[2] = bbox[3]
-        bbox[4] = tmp
+            bbox[3] = bbox[4]
+            bbox[4] = tmp
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes
 
 

tests/ut/python/dataset/test_resize_with_bbox.py

@@ -35,7 +35,11 @@ def fix_annotate(bboxes):
     :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
     """
     for (i, box) in enumerate(bboxes):
-        bboxes[i] = np.roll(box, -1)
+        if box.size == 7:
+            bboxes[i] = np.roll(box, -1)
+        else:
+            print("ERROR: Invalid Bounding Box size provided")
+            break
     return bboxes