rambo 11 miesięcy temu
rodzic
commit
e5b266c5b1
2 zmienionych plików z 2 dodań i 0 usunięć
  1. 1 0
      python/api.py
  2. 1 0
      python/services/deal_cutout.py

+ 1 - 0
python/api.py

@@ -62,6 +62,7 @@ async def segmentImages(params: SegmentImages):
     elif image_type == 0 and segment_type == 0:
         # 通用-普通抠图
         deal_cutout_mode = DealCutout(token)
+        deal_cutout_mode.need_cutout_images = need_cutout_images
         deal_cutout_mode.output_type = output_type
         result = deal_cutout_mode.normalMode()
     return success({"result": result})

+ 1 - 0
python/services/deal_cutout.py

@@ -52,6 +52,7 @@ class DealCutout:
         self.get_online_data.refresh_headers()
         num = 0
         result_array = []
+        print("self.need_cutout_images", self.need_cutout_images)
         for image_data in self.need_cutout_images:
             num += 1
             result = DealOneImageBeforehand(