fixed crop of single channel image like semantic map (#107)

27ce4190 · Jerry Jiarui XU · Kai Chen · 3b46ddf0 · 27ce4190
Commit 27ce4190 authored Aug 10, 2019 by Jerry Jiarui XU Committed by Kai Chen Aug 11, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/image/transforms/geometry.py mmcv/image/transforms/geometry.py +1 -1

No files found.
--- a/mmcv/image/transforms/geometry.py
+++ b/mmcv/image/transforms/geometry.py
@@ -142,7 +142,7 @@ def imcrop(img, bboxes, scale=1.0, pad_fill=None):
            patch = img[y1:y2 + 1, x1:x2 + 1, ...]
        else:
            _x1, _y1, _x2, _y2 = tuple(scaled_bboxes[i, :])
-            if chn == 2:
+            if chn == 1:
                patch_shape = (_y2 - _y1 + 1, _x2 - _x1 + 1)
            else:
                patch_shape = (_y2 - _y1 + 1, _x2 - _x1 + 1, chn)