xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix annotation

This commit is contained in:
xiefangqi 2020-07-21 12:04:56 +08:00
parent ad651f38bf
commit 64a6ed56de
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -252,9 +252,9 @@ def test_random_crop_with_bbox_op_bad_padding():
try: try:
test_op = c_vision.RandomCropWithBBox([512, 512], padding=-1) test_op = c_vision.RandomCropWithBBox([512, 512], padding=-1)
dataVoc2 = dataVoc2.map(input_columns=["image", "annotation"], dataVoc2 = dataVoc2.map(input_columns=["image", "bbox"],
output_columns=["image", "annotation"], output_columns=["image", "bbox"],
columns_order=["image", "annotation"], columns_order=["image", "bbox"],
operations=[test_op]) operations=[test_op])
for _ in dataVoc2.create_dict_iterator(): for _ in dataVoc2.create_dict_iterator():
@ -266,9 +266,9 @@ def test_random_crop_with_bbox_op_bad_padding():
try: try:
test_op = c_vision.RandomCropWithBBox([512, 512], padding=[16777216, 16777216, 16777216, 16777216]) test_op = c_vision.RandomCropWithBBox([512, 512], padding=[16777216, 16777216, 16777216, 16777216])
dataVoc2 = dataVoc2.map(input_columns=["image", "annotation"], dataVoc2 = dataVoc2.map(input_columns=["image", "bbox"],
output_columns=["image", "annotation"], output_columns=["image", "bbox"],
columns_order=["image", "annotation"], columns_order=["image", "bbox"],
operations=[test_op]) operations=[test_op])
for _ in dataVoc2.create_dict_iterator(): for _ in dataVoc2.create_dict_iterator():