feat 产品图打光模型部署

fix

app/service/utils/oss_client.py

@@ -0,0 +1,70 @@
+import io
+import logging
+from io import BytesIO
+
+import boto3
+import cv2
+import numpy as np
+from PIL import Image
+from minio import Minio
+
+from app.core.config import *
+
+logger = logging.getLogger()
+
+
+#  获取图片
+def oss_get_image(bucket, object_name, data_type):
+    image_object = None
+
+    try:
+        if OSS == "minio":
+            oss_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
+            image_data = oss_client.get_object(bucket_name=bucket, object_name=object_name)
+        else:
+            oss_client = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY, region_name=S3_REGION_NAME)
+            image_data = oss_client.get_object(Bucket=bucket, Key=object_name)['Body']
+
+        if data_type == "cv2":
+            image_bytes = image_data.read()
+            image_array = np.frombuffer(image_bytes, np.uint8)  # 转成8位无符号整型
+            image_object = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
+        else:
+            data_bytes = BytesIO(image_data.read())
+            image_object = Image.open(data_bytes)
+    except Exception as e:
+        logger.warning(f"{OSS} | 获取图片出现异常 ######: {e}")
+    return image_object
+
+
+def oss_upload_image(bucket, object_name, image_bytes):
+    req = None
+    try:
+        if OSS == "minio":
+            oss_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
+            req = oss_client.put_object(bucket_name=bucket, object_name=object_name, data=io.BytesIO(image_bytes), length=len(image_bytes), content_type='image/png')
+        else:
+            oss_client = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY, region_name=S3_REGION_NAME)
+            req = oss_client.put_object(Bucket=AIDA_CLOTHING, Key=object_name, Body=image_bytes, ContentType='image/png')
+    except Exception as e:
+        logger.warning(f"{OSS} | 上传图片出现异常 ######: {e}")
+    return req
+
+
+if __name__ == '__main__':
+    # url = "aida-results/result_0002186a-e631-11ee-86a6-b48351119060.png"
+    # url = "aida-collection-element/11523/Moodboard/f60af0d2-94c2-48f9-90ff-74b8e8a481b5.jpg"
+    # url = "aida-sys-image/images/female/outwear/0628000054.jpg"
+    # url = "aida-users/89/product_image/string-89.png"
+    # url = "aida-users/89/single_logo/123-89.png"
+    # url = 'aida-users/89/relight_image/123-89.png'
+    # url = 'aida-users/89/relight_image/123-89.png'
+    url = 'aida-users/89/relight_image/123-89.png'
+    read_type = "PIL"
+    if read_type == "cv2":
+        img = oss_get_image(bucket=url.split('/')[0], object_name=url[url.find('/') + 1:], data_type=read_type)
+        cv2.imshow("", img)
+        cv2.waitKey(0)
+    else:
+        img = oss_get_image(bucket=url.split('/')[0], object_name=url[url.find('/') + 1:], data_type=read_type)
+        img.show()