Update main_test_CMFNet.py

main_test_CMFNet.py

@@ -42,7 +42,7 @@ def main():
     load_checkpoint(model, args.weights)
     
 
-    mul = 16
+    mul = 8
     for file_ in files:
         img = Image.open(file_).convert('RGB')
         input_ = TF.to_tensor(img).unsqueeze(0).to(device)
@@ -55,7 +55,7 @@ def main():
         input_ = F.pad(input_, (0, padw, 0, padh), 'reflect')
         with torch.no_grad():
             restored = model(input_)
-
+        restored = restored[0]
         restored = torch.clamp(restored, 0, 1)
         restored = restored[:, :, :h, :w]
         restored = restored.permute(0, 2, 3, 1).cpu().detach().numpy()