Update main_test_CMFNet.py

main_test_CMFNet.py

@@ -53,10 +53,11 @@ def main():
         padh = H - h if h % mul != 0 else 0
         padw = W - w if w % mul != 0 else 0
         input_ = F.pad(input_, (0, padw, 0, padh), 'reflect')
+        
         with torch.no_grad():
             restored = model(input_)
             
-        restored = torch.clamp(restored[1], 0, 1)
+        restored = torch.clamp(restored, 0, 1)
         restored = restored[:, :, :h, :w]
         restored = restored.permute(0, 2, 3, 1).cpu().detach().numpy()
         restored = img_as_ubyte(restored[0])