From 64a6ed56deca686522c54e304a697d5116153d4b Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Tue, 21 Jul 2020 12:04:56 +0800 Subject: [PATCH] fix annotation --- .../ut/python/dataset/test_random_crop_with_bbox.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/ut/python/dataset/test_random_crop_with_bbox.py b/tests/ut/python/dataset/test_random_crop_with_bbox.py index 97a8d2ee866..28a68a7c38a 100644 --- a/tests/ut/python/dataset/test_random_crop_with_bbox.py +++ b/tests/ut/python/dataset/test_random_crop_with_bbox.py @@ -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():