Update 1removeMask.py

1removeMask.py

@@ -122,6 +122,17 @@ def save_img(img_content: "要处理的图片数据", new_save_path: "新文件
         os.remove(old_img_path)
 
 
+# 传入遮罩图片路径和需要去水印的图片路径,将调整大小后的mask保存 ->void
+def mask_resize(maskPath:str,removeMarkImagePath:str):
+    maskImg = Image.open(maskPath)
+    # 定义新的图片大小（宽度，高度）
+    new_size = Image.open(removeMarkImagePath).size  # 例如，将图片调整为宽400像素，高300像素
+    print("遮罩大小是:", maskImg.size, "不匹配图片大小是:", new_size,"不匹配图片路径是",removeMarkImagePath)
+    # 调整图片大小
+    resized_img = maskImg.resize(new_size)
+    # 保存调整大小后的图片
+    resized_img.save(maskPath)
+
 if __name__ == '__main__':
     main(FakeArgs())#初始化model
     # 获取当前目录的子目录的路径
@@ -151,7 +162,9 @@ if __name__ == '__main__':
 
         if not os.path.exists(new_img_path):
             # 如果已经处理过那么跳过
-            # 处理图片并保存
+            # 处理图片并保存 ->每次处理请求的时候都要调整mask大小，使其和img大小一致,这样就可以在taskManger的时候减轻判断负担
+            mask_resize(maskPath ='mask/0.jpg', removeMarkImagePath=img_file)
+
             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: