[Fix] fix ssim only calculate on the first channel bug (#1515)

open-mmlab · Dec 19, 2022 · b0f5592 · b0f5592
1 parent ccd52c4
commit b0f5592
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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