mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity

ImgAug: ignores->ignored

Browse Source
This commit is contained in:
gaotongxiao 2022-05-19 15:31:52 +08:00
parent 178030bad6
commit 79186b61ec
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mmocr/datasets/pipelines/wrappers.py
View File

@ -22,7 +22,7 @@ class ImgAug(BaseTransform):
- gt_polygons (optional for text recognition) - gt_polygons (optional for text recognition)
- gt_bboxes (optional for text recognition) - gt_bboxes (optional for text recognition)
- gt_bboxes_labels (optional for text recognition) - gt_bboxes_labels (optional for text recognition)
- gt_ignores (optional for text recognition) - gt_ignored (optional for text recognition)
- gt_texts (optional) - gt_texts (optional)
Modified Keys: Modified Keys:
@ -31,7 +31,7 @@ class ImgAug(BaseTransform):
- gt_polygons (optional for text recognition) - gt_polygons (optional for text recognition)
- gt_bboxes (optional for text recognition) - gt_bboxes (optional for text recognition)
- gt_bboxes_labels (optional for text recognition) - gt_bboxes_labels (optional for text recognition)
- gt_ignores (optional for text recognition) - gt_ignored (optional for text recognition)
- img_shape (optional) - img_shape (optional)
- gt_texts (optional) - gt_texts (optional)
@ -91,7 +91,7 @@ class ImgAug(BaseTransform):
Returns: Returns:
dict: The transformed data. dict: The transformed data.
""" """
# Assume co-existence of `gt_polygons`, `gt_bboxes` and `gt_ignores` # Assume co-existence of `gt_polygons`, `gt_bboxes` and `gt_ignored`
# for text detection # for text detection
if 'gt_polygons' in results: if 'gt_polygons' in results:
@ -104,8 +104,8 @@ class ImgAug(BaseTransform):
results['gt_bboxes'], removed_poly_inds, axis=0) results['gt_bboxes'], removed_poly_inds, axis=0)
results['gt_bboxes_labels'] = np.delete( results['gt_bboxes_labels'] = np.delete(
results['gt_bboxes_labels'], removed_poly_inds, axis=0) results['gt_bboxes_labels'], removed_poly_inds, axis=0)
results['gt_ignores'] = np.delete( results['gt_ignored'] = np.delete(
results['gt_ignores'], removed_poly_inds, axis=0) results['gt_ignored'], removed_poly_inds, axis=0)
# TODO: deal with gt_texts corresponding to clipped polygons # TODO: deal with gt_texts corresponding to clipped polygons
if 'gt_texts' in results: if 'gt_texts' in results:
results['gt_texts'] = [ results['gt_texts'] = [

20
tests/test_datasets/test_pipelines/test_wrappers.py
View File

@ -25,7 +25,7 @@ class TestImgAug(unittest.TestCase):
# It shall always be 0 in MMOCR, but we assign different labels to # It shall always be 0 in MMOCR, but we assign different labels to
# dummy instances for testing # dummy instances for testing
labels = np.array([0, 1], dtype=np.int64) labels = np.array([0, 1], dtype=np.int64)
ignores = np.array([False, True], dtype=bool) ignored = np.array([False, True], dtype=bool)
texts = ['text1', 'text2'] texts = ['text1', 'text2']
return dict( return dict(
img=img, img=img,
@ -33,7 +33,7 @@ class TestImgAug(unittest.TestCase):
gt_polygons=poly, gt_polygons=poly,
gt_bboxes=box, gt_bboxes=box,
gt_bboxes_labels=labels, gt_bboxes_labels=labels,
gt_ignores=ignores, gt_ignored=ignored,
gt_texts=texts) gt_texts=texts)
def assertPolyEqual(self, poly1: List[np.ndarray], def assertPolyEqual(self, poly1: List[np.ndarray],
@ -53,7 +53,7 @@ class TestImgAug(unittest.TestCase):
self.assertTrue(np.array_equal(bbox_targets, results['gt_bboxes'])) self.assertTrue(np.array_equal(bbox_targets, results['gt_bboxes']))
self.assertTrue( self.assertTrue(
np.array_equal(bbox_label_targets, results['gt_bboxes_labels'])) np.array_equal(bbox_label_targets, results['gt_bboxes_labels']))
self.assertTrue(np.array_equal(ignore_targets, results['gt_ignores'])) self.assertTrue(np.array_equal(ignore_targets, results['gt_ignored']))
self.assertEqual(text_targets, results['gt_texts']) self.assertEqual(text_targets, results['gt_texts'])
self.assertEqual(results['img_shape'], self.assertEqual(results['img_shape'],
(results['img'].shape[0], results['img'].shape[1])) (results['img'].shape[0], results['img'].shape[1]))
@ -68,7 +68,7 @@ class TestImgAug(unittest.TestCase):
self.assert_result_equal(results, origin_results['gt_polygons'], self.assert_result_equal(results, origin_results['gt_polygons'],
origin_results['gt_bboxes'], origin_results['gt_bboxes'],
origin_results['gt_bboxes_labels'], origin_results['gt_bboxes_labels'],
origin_results['gt_ignores'], origin_results['gt_ignored'],
origin_results['gt_texts']) origin_results['gt_texts'])
args = [dict(cls='Affine', translate_px=dict(x=-10, y=-10))] args = [dict(cls='Affine', translate_px=dict(x=-10, y=-10))]
@ -84,10 +84,10 @@ class TestImgAug(unittest.TestCase):
] ]
box_target = np.array([[0, 0, 40, 40], [10, 10, 40, 40]]) box_target = np.array([[0, 0, 40, 40], [10, 10, 40, 40]])
label_target = np.array([0, 1], dtype=np.int64) label_target = np.array([0, 1], dtype=np.int64)
ignores = np.array([False, True], dtype=bool) ignored = np.array([False, True], dtype=bool)
texts = ['text1', 'text2'] texts = ['text1', 'text2']
self.assert_result_equal(results, poly_target, box_target, self.assert_result_equal(results, poly_target, box_target,
label_target, ignores, texts) label_target, ignored, texts)
# Some polygons and bboxes are no longer inside the image after # Some polygons and bboxes are no longer inside the image after
# transformation # transformation
@ -97,13 +97,13 @@ class TestImgAug(unittest.TestCase):
poly_target = [np.array([0, 30, 20, 30, 20, 50, 0, 50])] poly_target = [np.array([0, 30, 20, 30, 20, 50, 0, 50])]
box_target = np.array([[0, 30, 20, 50]]) box_target = np.array([[0, 30, 20, 50]])
label_target = np.array([0], dtype=np.int64) label_target = np.array([0], dtype=np.int64)
ignores = np.array([False], dtype=bool) ignored = np.array([False], dtype=bool)
texts = ['text1'] texts = ['text1']
imgaug_transform = ImgAug(args) imgaug_transform = ImgAug(args)
results = self._create_dummy_data() results = self._create_dummy_data()
results = imgaug_transform(results) results = imgaug_transform(results)
self.assert_result_equal(results, poly_target, box_target, self.assert_result_equal(results, poly_target, box_target,
label_target, ignores, texts) label_target, ignored, texts)
# All polygons and bboxes are no longer inside the image after # All polygons and bboxes are no longer inside the image after
# transformation # transformation
@ -111,13 +111,13 @@ class TestImgAug(unittest.TestCase):
poly_target = [] poly_target = []
box_target = np.zeros((0, 4)) box_target = np.zeros((0, 4))
label_target = np.array([], dtype=np.int64) label_target = np.array([], dtype=np.int64)
ignores = np.array([], dtype=bool) ignored = np.array([], dtype=bool)
texts = [] texts = []
imgaug_transform = ImgAug(args) imgaug_transform = ImgAug(args)
results = self._create_dummy_data() results = self._create_dummy_data()
results = imgaug_transform(results) results = imgaug_transform(results)
self.assert_result_equal(results, poly_target, box_target, self.assert_result_equal(results, poly_target, box_target,
label_target, ignores, texts) label_target, ignored, texts)
# Everything should work well without gt_texts # Everything should work well without gt_texts
results = self._create_dummy_data() results = self._create_dummy_data()