From dcfedf54f1fbfa30b10a62ab20f407be8bee5b66 Mon Sep 17 00:00:00 2001
From: Cao Yuhang <yhcao6@gmail.com>
Date: Fri, 27 Dec 2019 19:57:44 +0800
Subject: [PATCH] Add Crop seg to RandomCrop (#1880)
* add crop seg to RandomCrop trans
* minor fix comment
---
mmdet/datasets/pipelines/transforms.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py
index f57eb4e..702bdea 100644
--- a/mmdet/datasets/pipelines/transforms.py
+++ b/mmdet/datasets/pipelines/transforms.py
@@ -378,6 +378,10 @@ class RandomCrop(object):
bboxes[:, 1::2] = np.clip(bboxes[:, 1::2], 0, img_shape[0] - 1)
results[key] = bboxes
+ # crop semantic seg
+ for key in results.get('seg_fields', []):
+ results[key] = results[key][crop_y1:crop_y2, crop_x1:crop_x2]
+
# filter out the gt bboxes that are completely cropped
if 'gt_bboxes' in results:
gt_bboxes = results['gt_bboxes']
--
GitLab