feat design 提速测试

fix

app/service/design_test/utils/upload_image.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+"""
+@Project ：trinity_client
+@File    ：upload_image.py
+@Author  ：周成融
+@Date    ：2023/8/28 13:49:20
+@detail  ：
+"""
+import io
+import logging
+
+import cv2
+
+from app.core.config import *
+from app.service.utils.decorator import RunTime
+from app.service.utils.oss_client import oss_upload_image
+
+
+# @RunTime
+def upload_png_mask(front_image, object_name, mask=None):
+    try:
+        mask_url = None
+        if mask is not None:
+            mask_inverted = cv2.bitwise_not(mask)
+            # 将掩模的3通道转换为4通道，白色部分不透明，黑色部分透明
+            rgba_image = cv2.cvtColor(mask_inverted, cv2.COLOR_BGR2BGRA)
+            rgba_image[rgba_image[:, :, 0] == 0] = [0, 0, 0, 0]
+            # image_bytes = io.BytesIO()
+            # image_bytes.write(cv2.imencode('.png', rgba_image)[1].tobytes())
+            # image_bytes.seek(0)
+            # mask_url = f"{AIDA_CLOTHING}/{minio_client.put_object('aida-clothing', f'mask/mask_{object_name}.png', image_bytes, len(image_bytes.getvalue()), content_type='image/png').object_name}"
+            # oss upload ####################
+            req = oss_upload_image(bucket=AIDA_CLOTHING, object_name=f"mask/mask_{object_name}.png", image_bytes=cv2.imencode('.png', rgba_image)[1])
+            mask_url = f"{AIDA_CLOTHING}/mask/mask_{object_name}.png"
+
+        image_data = io.BytesIO()
+        front_image.save(image_data, format='PNG')
+        image_data.seek(0)
+        image_bytes = image_data.read()
+        # image_url = f"{AIDA_CLOTHING}/{minio_client.put_object('aida-clothing', f'image/image_{object_name}.png', io.BytesIO(image_bytes), len(image_bytes), content_type='image/png').object_name}"
+        req = oss_upload_image(bucket=AIDA_CLOTHING, object_name=f"image/image_{object_name}.png", image_bytes=image_bytes)
+        image_url = f"{AIDA_CLOTHING}/image/image_{object_name}.png"
+        return front_image, image_url, mask_url
+    except Exception as e:
+        logging.warning(f"upload_png_mask runtime exception : {e}")
+
+
+# @RunTime
+# def upload_png_mask(front_image, object_name, mask=None):
+#     mask_url = None
+#     if mask is not None:
+#         mask_url = f"{AIDA_CLOTHING}/mask/mask_{object_name}.png"
+#     image_url = f"{AIDA_CLOTHING}/image/image_{object_name}.png"
+#     return front_image, image_url, mask_url