fix lpips.py

XPixelGroup · Nov 10, 2021 · 10f55c8 · 10f55c8
1 parent a74fb57
commit 10f55c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/basicsr/metrics/lpips.py b/basicsr/metrics/lpips.py
@@ -8,7 +8,7 @@
 import torch
 
 @METRIC_REGISTRY.register()
-def calculate_lpips(img, img2, crop_border, input_order='HWC', test_y_channel=False):
+def calculate_lpips(img, img2, crop_border, input_order='HWC', test_y_channel=False, **kwargs):
 
     """Calculate LPIPS.
     Ref: https://github.com/xinntao/BasicSR/pull/367