From 374edce9ef2b9a2692781b64fe9ee1ed39aee464 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 24 Jun 2024 17:39:29 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=E7=94=B1=E4=BA=8E=E5=BB=B6?= =?UTF-8?q?=E8=BF=9F=EF=BC=8Cbounding=20box=E5=90=8E=E7=9A=84sketch?= =?UTF-8?q?=E4=B8=8E=E5=89=8D=E7=AB=AF=E7=BC=93=E5=AD=98=E7=9A=84sketch?= =?UTF-8?q?=E6=9C=89=E8=AF=AF=E5=B7=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_pre_processing/service.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index f6f239a..272dc7e 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -9,6 +9,7 @@ from urllib3.exceptions import ResponseError from app.core.config import * from app.service.design.utils.design_ensemble import get_keypoint_result +from app.service.utils.generate_uuid import generate_uuid from app.service.utils.oss_client import oss_get_image, oss_upload_image @@ -121,8 +122,11 @@ class DesignPreprocessing: image_bytes = cv2.imencode(".jpg", item['obj'])[1].tobytes() # self.minio_client.put_object(item['image_url'].split("/", 1)[0], item['image_url'].split("/", 1)[1], io.BytesIO(image_bytes), len(image_bytes), content_type="image/jpeg", ) bucket_name = item['image_url'].split("/", 1)[0] + # 由于延迟，bounding box后的sketch与前端缓存的sketch有误差 object_name = item['image_url'].split("/", 1)[1] - oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) + new_object = f"{object_name[:object_name.rfind('/') + 1]}{generate_uuid()}.{object_name.split('.', 1)[1]}" + oss_upload_image(bucket=bucket_name, object_name=new_object, image_bytes=image_bytes) + item['new_image_url'] = f"{bucket_name}/{new_object}" print(f"Object '{item['image_url'].split('/', 1)[1]}' overwritten successfully.") except ResponseError as err: print(f"Error: {err}")