mlbench123 commited on
Commit
027f7d3
·
verified ·
1 Parent(s): d8915bc

Update apply_mask.py

Browse files
Files changed (1) hide show
  1. apply_mask.py +1 -1
apply_mask.py CHANGED
@@ -68,7 +68,7 @@ def apply_mask_and_crop(input_folder, mask, output_folder):
68
 
69
  # Save cropped image
70
  cv2.imwrite(output_path, cropped_img)
71
- if i = 0:
72
  preview_path = os.path.join(output_folder, f"preview_{filename}")
73
  cv2.imwrite(preview_path, cropped_img)
74
  return preview_path
 
68
 
69
  # Save cropped image
70
  cv2.imwrite(output_path, cropped_img)
71
+ if i == 0:
72
  preview_path = os.path.join(output_folder, f"preview_{filename}")
73
  cv2.imwrite(preview_path, cropped_img)
74
  return preview_path