d

python/services/deal_cutout.py

@@ -1,7 +1,7 @@
 import time
 from concurrent.futures import as_completed, ThreadPoolExecutor, wait
 import threading
-from .remove_bg_pixian import RemoveBgPiXian
+from .remove_bg_pixian import RemoveBgPiXian,Picture
 from .other.module_online_data import GetOnlineData
 from .deal_one_image import DealOneImage, DealOneImageBeforehand
 from .other.log import MyLogger
@@ -10,7 +10,7 @@ import pandas as pd
 import csv
 from PIL import Image
 from io import BytesIO
-import os,requests
+import os,requests,io
 def urlPilImage(url):
     yzmdata = requests.get(url)
     tempIm = BytesIO(yzmdata.content)
@@ -151,11 +151,22 @@ class DealModelForm:
         num = 0
         save_root_path = ""
         baseImages = []
+        resize=1600#定义标准
         for index,image_data in enumerate(self.need_cutout_images):
             num += 1
             save_root_path = image_data["root_path"]
             file_path = image_data["file_path"]
-            image_url = self.get_online_data.upload_pic(file_path)
+            original_pic = Picture(file_path)
+            if original_pic.x > original_pic.y:
+                if original_pic.x > resize:
+                    original_pic.resize(resize)
+            else:
+                if original_pic.y > resize:
+                    original_pic.resize_by_heigh(heigh=resize)
+            buffer = io.BytesIO()
+            original_pic.im.save(buffer, format='JPEG')
+            buffer.seek(0)
+            image_url = self.get_online_data.upload_pic(file_path=None,buffer=buffer)
             baseImages.append(image_url)
             self.need_cutout_images[index]["image_url"] = image_url
         data = {