Browse Source

!3263 Fix bbox interface annotation column name to bbox

Merge pull request !3263 from xiefangqi/md_fix_annotation
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
25e587e483
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      tests/ut/python/dataset/test_random_crop_with_bbox.py

+ 6
- 6
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:
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…
Cancel
Save