Update 1removeMask.py

1removeMask.py

@@ -101,7 +101,7 @@ if __name__ == '__main__':
         if not os.path.exists(new_img_path):
             #如果已经处理过那么跳过
             # 处理图片并保存
-            img_inpainted = inpaint(img_path=img_file, mask_path='mask/all_mask.jpg')
+            img_inpainted = inpaint(img_path=img_file, mask_path='mask/0.jpg')#上传的遮罩保存都是0开始
             save_img(img_content=img_inpainted, new_save_path=new_img_path,old_img_path=img_file)
         else:
             print(f"Skipping {new_img_path} as it already exists.")