Update inference_img.py

inference_img.py

@@ -100,7 +100,18 @@ import warnings
     for i in range(len(img_list)):
         if args.img[0].endswith('.exr') and args.img[1].endswith('.exr'):
             cv2.imwrite('output/img{}.exr'.format(i), (img_list[i][0]).cpu().numpy().transpose(1, 2, 0)[:h, :w], [cv2.IMWRITE_EXR_TYPE, cv2.IMWRITE_EXR_TYPE_HALF])
+
+            # Replace this line (or add below your current cv2.imwrite)
+            save_path = 'output/img{}.png'.format(i)
+            success = cv2.imwrite(save_path, (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
+            print(f"Saving to {save_path} → success: {success}")
+
         else:
             cv2.imwrite('output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
+            
+            # Replace this line (or add below your current cv2.imwrite)
+            save_path = 'output/img{}.png'.format(i)
+            success = cv2.imwrite(save_path, (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
+            print(f"Saving to {save_path} → success: {success}")