forked from mindspore-Ecosystem/mindspore
!3263 Fix bbox interface annotation column name to bbox
Merge pull request !3263 from xiefangqi/md_fix_annotation
This commit is contained in:
commit
25e587e483
|
@ -252,9 +252,9 @@ def test_random_crop_with_bbox_op_bad_padding():
|
|||
try:
|
||||
test_op = c_vision.RandomCropWithBBox([512, 512], padding=-1)
|
||||
|
||||
dataVoc2 = dataVoc2.map(input_columns=["image", "annotation"],
|
||||
output_columns=["image", "annotation"],
|
||||
columns_order=["image", "annotation"],
|
||||
dataVoc2 = dataVoc2.map(input_columns=["image", "bbox"],
|
||||
output_columns=["image", "bbox"],
|
||||
columns_order=["image", "bbox"],
|
||||
operations=[test_op])
|
||||
|
||||
for _ in dataVoc2.create_dict_iterator():
|
||||
|
@ -266,9 +266,9 @@ def test_random_crop_with_bbox_op_bad_padding():
|
|||
try:
|
||||
test_op = c_vision.RandomCropWithBBox([512, 512], padding=[16777216, 16777216, 16777216, 16777216])
|
||||
|
||||
dataVoc2 = dataVoc2.map(input_columns=["image", "annotation"],
|
||||
output_columns=["image", "annotation"],
|
||||
columns_order=["image", "annotation"],
|
||||
dataVoc2 = dataVoc2.map(input_columns=["image", "bbox"],
|
||||
output_columns=["image", "bbox"],
|
||||
columns_order=["image", "bbox"],
|
||||
operations=[test_op])
|
||||
|
||||
for _ in dataVoc2.create_dict_iterator():
|
||||
|
|
Loading…
Reference in New Issue