Add Crop seg to RandomCrop (#1880)

* add crop seg to RandomCrop trans * minor fix comment

Add Crop seg to RandomCrop (#1880)
* add crop seg to RandomCrop trans * minor fix comment
dcfedf54 · Cao Yuhang · Kai Chen · 16a6f7da · dcfedf54
Commit dcfedf54 authored Dec 27, 2019 by Cao Yuhang Committed by Kai Chen Dec 27, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

mmdet/datasets/pipelines/transforms.py mmdet/datasets/pipelines/transforms.py +4 -0

No files found.
--- 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']