yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!2853 BBox Ops Python Test file - Fix_Annotate function bug fix 

Merge pull request !2853 from danishnxt/AugOps-fix
This commit is contained in:
mindspore-ci-bot 2020-07-04 00:41:38 +08:00 committed by Gitee
parent f2a99538fd 5a9673ee6f
commit 7cc4b8438e
7 changed files with 60 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/ut/python/dataset/test_bounding_box_augment.py
View File

@ -34,12 +34,16 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for bbox in bboxes: 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 return bboxes

16
tests/ut/python/dataset/test_random_crop_and_resize_with_bbox.py
View File

@ -36,12 +36,16 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for bbox in bboxes: 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 return bboxes

16
tests/ut/python/dataset/test_random_crop_with_bbox.py
View File

@ -37,12 +37,16 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for bbox in bboxes: 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 return bboxes

16
tests/ut/python/dataset/test_random_horizontal_flip_with_bbox.py
View File

@ -34,12 +34,16 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for bbox in bboxes: 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 return bboxes

6
tests/ut/python/dataset/test_random_resize_with_bbox.py
View File

@ -35,7 +35,11 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for (i, box) in enumerate(bboxes): 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 return bboxes

16
tests/ut/python/dataset/test_random_vertical_flip_with_bbox.py
View File

@ -36,12 +36,16 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for bbox in bboxes: 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 return bboxes

6
tests/ut/python/dataset/test_resize_with_bbox.py
View File

@ -35,7 +35,11 @@ def fix_annotate(bboxes):
:return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format :return: annotation in [x_min, y_min, w, h, label, truncate, difficult] format
""" """
for (i, box) in enumerate(bboxes): 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 return bboxes