aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat generate sketch 新增 服装类别识别

Browse Source
This commit is contained in:
zhouchengrong
2024-04-25 10:48:32 +08:00
parent 3cdeb32443
commit 6f27a99e43
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/service/generate_image/utils/upload_sd_image.py
View File

@@ -10,6 +10,7 @@
import io import io
import logging import logging
import cv2
from PIL import Image from PIL import Image
from minio import Minio from minio import Minio
@@ -20,15 +21,12 @@ minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET
def upload_png_sd(image, user_id, category, object_name): def upload_png_sd(image, user_id, category, object_name):
try: try:
image_file = io.BytesIO() _, img_byte_array = cv2.imencode('.jpg', image)
image = Image.fromarray(image)
image.save(image_file, format='JPEG')
image_file.seek(0)
minio_req = minio_client.put_object( minio_req = minio_client.put_object(
GI_MINIO_BUCKET, GI_MINIO_BUCKET,
f'{user_id}/{category}/{object_name}', f'{user_id}/{category}/{object_name}',
image_file, io.BytesIO(img_byte_array),
len(image_file.getvalue()), len(img_byte_array),
content_type='image/jpeg' content_type='image/jpeg'
) )
image_url = f"aida-users/{minio_req.object_name}" image_url = f"aida-users/{minio_req.object_name}"