diff --git a/mmedit/evaluation/metrics/metrics_utils.py b/mmedit/evaluation/metrics/metrics_utils.py index 4ed5602a64..64630454f4 100644 --- a/mmedit/evaluation/metrics/metrics_utils.py +++ b/mmedit/evaluation/metrics/metrics_utils.py @@ -121,7 +121,7 @@ def img_transform(img, '"Y" and None.') if crop_border != 0: - img = img[crop_border:-crop_border, crop_border:-crop_border, None] + img = img[crop_border:-crop_border, crop_border:-crop_border, ...] return img