feat

fix design 多进程处理
2024-07-25 10:57:59 +08:00
parent 8dd6fc924c
commit 9093eaf674
2 changed files with 37 additions and 36 deletions
--- a/app/service/design/service.py
+++ b/app/service/design/service.py
@@ -1,5 +1,6 @@
 import concurrent.futures
 import io
+from concurrent.futures import ProcessPoolExecutor

 import cv2

@@ -62,7 +63,7 @@ def generate(request_data):
            obj = futures[future]
            return_response[obj] = future.result()[0]
            return_png_mask.extend(future.result()[1])
-    # upload_results = process_images(return_png_mask)
+    upload_results = process_images(return_png_mask)
    final_progress(process_id)
    return return_response

@@ -151,7 +152,7 @@ def process_object(cfg, process_id, total):

@RunTime
 def process_images(images):
-    with concurrent.futures.ThreadPoolExecutor() as executor:
+    with ProcessPoolExecutor() as executor:
        results = list(executor.map(upload_images, images))
    # results = []
    # for image in images: