From 1c9afecf2d08a38aeae1a488621ecdaccb79bea0 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Wed, 27 Nov 2019 22:01:38 +0800
Subject: [PATCH] recover PR#1115 to add flip direction (#1273)
---
mmdet/datasets/pipelines/transforms.py | 33 +++++++++++++++++++-------
1 file changed, 25 insertions(+), 8 deletions(-)
diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py
index fe1584d..e6d3b68 100644
--- a/mmdet/datasets/pipelines/transforms.py
+++ b/mmdet/datasets/pipelines/transforms.py
@@ -179,12 +179,14 @@ class RandomFlip(object):
flip_ratio (float, optional): The flipping probability.
"""
- def __init__(self, flip_ratio=None):
+ def __init__(self, flip_ratio=None, direction='horizontal'):
self.flip_ratio = flip_ratio
+ self.direction = direction
if flip_ratio is not None:
assert flip_ratio >= 0 and flip_ratio <= 1
+ assert direction in ['horizontal', 'vertical']
- def bbox_flip(self, bboxes, img_shape):
+ def bbox_flip(self, bboxes, img_shape, direction):
"""Flip bboxes horizontally.
Args:
@@ -192,26 +194,41 @@ class RandomFlip(object):
img_shape(tuple): (height, width)
"""
assert bboxes.shape[-1] % 4 == 0
- w = img_shape[1]
flipped = bboxes.copy()
- flipped[..., 0::4] = w - bboxes[..., 2::4] - 1
- flipped[..., 2::4] = w - bboxes[..., 0::4] - 1
+ if direction == 'horizontal':
+ w = img_shape[1]
+ flipped[..., 0::4] = w - bboxes[..., 2::4] - 1
+ flipped[..., 2::4] = w - bboxes[..., 0::4] - 1
+ elif direction == 'vertical':
+ h = img_shape[0]
+ flipped[..., 1::4] = h - bboxes[..., 3::4] - 1
+ flipped[..., 3::4] = h - bboxes[..., 1::4] - 1
+ else:
+ raise ValueError(
+ 'Invalid flipping direction "{}"'.format(direction))
return flipped
def __call__(self, results):
if 'flip' not in results:
flip = True if np.random.rand() < self.flip_ratio else False
results['flip'] = flip
+ if 'flip_direction' not in results:
+ results['flip_direction'] = self.direction
if results['flip']:
# flip image
- results['img'] = mmcv.imflip(results['img'])
+ results['img'] = mmcv.imflip(
+ results['img'], direction=results['flip_direction'])
# flip bboxes
for key in results.get('bbox_fields', []):
results[key] = self.bbox_flip(results[key],
- results['img_shape'])
+ results['img_shape'],
+ results['flip_direction'])
# flip masks
for key in results.get('mask_fields', []):
- results[key] = [mask[:, ::-1] for mask in results[key]]
+ results[key] = [
+ mmcv.imflip(mask, direction=results['flip_direction'])
+ for mask in results[key]
+ ]
return results
def __repr__(self):
--
GitLab