Update inference_img.py

inference_img.py

@@ -114,3 +114,4 @@ for i in range(len(img_list)):
         success = cv2.imwrite(filename_png, (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
         print(f"Saving to {filename_png} → success: {success}")
         print("Saving to:", os.path.abspath(filename_png))
+