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

feat

Browse Source 
fix 由于延迟，bounding box后的sketch与前端缓存的sketch有误差
This commit is contained in:
zhouchengrong
2024-06-24 17:39:29 +08:00
parent 2a83399eff
commit 374edce9ef
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/service/design_pre_processing/service.py
View File

@@ -9,6 +9,7 @@ from urllib3.exceptions import ResponseError
from app.core.config import * from app.core.config import *
from app.service.design.utils.design_ensemble import get_keypoint_result 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 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() 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", ) # 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] bucket_name = item['image_url'].split("/", 1)[0]
# 由于延迟，bounding box后的sketch与前端缓存的sketch有误差
object_name = item['image_url'].split("/", 1)[1] 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.") print(f"Object '{item['image_url'].split('/', 1)[1]}' overwritten successfully.")
except ResponseError as err: except ResponseError as err:
print(f"Error: {err}") print(f"Error: {err}")