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

feat generate 污点图保存

Browse Source
This commit is contained in:
zhouchengrong
2024-05-13 11:35:27 +08:00
parent cce6e987d7
commit bb46282987
3 changed files with 57 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -33,3 +33,35 @@ def upload_png_sd(image, user_id, category, object_name):
return image_url
except Exception as e:
logging.warning(f"upload_png_mask runtime exception : {e}")
def upload_stain_png_sd(image, user_id, category, object_name):
try:
_, img_byte_array = cv2.imencode('.jpg', image)
minio_req = minio_client.put_object(
"test",
f'generate_result/stain/{user_id}_{category}_{object_name}',
io.BytesIO(img_byte_array),
len(img_byte_array),
content_type='image/jpeg'
)
image_url = f"test/{minio_req.object_name}"
return image_url
except Exception as e:
logging.warning(f"upload_png_mask runtime exception : {e}")
def upload_face_png_sd(image, user_id, category, object_name):
try:
_, img_byte_array = cv2.imencode('.jpg', image)
minio_req = minio_client.put_object(
"test",
f'generate_result/face/{user_id}_{category}_{object_name}',
io.BytesIO(img_byte_array),
len(img_byte_array),
content_type='image/jpeg'
)
image_url = f"test/{minio_req.object_name}"
return image_url
except Exception as e:
logging.warning(f"upload_png_mask runtime exception : {e}")