From 6f27a99e43264fd4310611281c679617e8a4e0fb Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 25 Apr 2024 10:48:32 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20sketch=20=E6=96=B0=E5=A2=9E?= =?UTF-8?q?=20=E6=9C=8D=E8=A3=85=E7=B1=BB=E5=88=AB=E8=AF=86=E5=88=AB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/utils/upload_sd_image.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/app/service/generate_image/utils/upload_sd_image.py b/app/service/generate_image/utils/upload_sd_image.py index c6a0632..0c3baec 100644 --- a/app/service/generate_image/utils/upload_sd_image.py +++ b/app/service/generate_image/utils/upload_sd_image.py @@ -10,6 +10,7 @@ import io import logging +import cv2 from PIL import Image 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): try: - image_file = io.BytesIO() - image = Image.fromarray(image) - image.save(image_file, format='JPEG') - image_file.seek(0) + _, img_byte_array = cv2.imencode('.jpg', image) minio_req = minio_client.put_object( GI_MINIO_BUCKET, f'{user_id}/{category}/{object_name}', - image_file, - len(image_file.getvalue()), + io.BytesIO(img_byte_array), + len(img_byte_array), content_type='image/jpeg' ) image_url = f"aida-users/{minio_req.object_name}"