Update model/CMFNet.py

model/CMFNet.py

@@ -188,6 +188,6 @@ class CMFNet(nn.Module):
         concat_feat = self.concat123(torch.cat([x1_out, x2_out, x3_out], 1))
         x_final = self.tail(concat_feat)
 
-        return [x_final + mixed_img, mixed_img, mix_r[1], x1_img, x2_img, x3_img, x_final]
+        return x_final + mixed_img