diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index 88ed739..f6f239a 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -4,12 +4,8 @@ import time import cv2 import numpy as np import torch -from minio import Minio -from pymilvus import connections, Collection -from urllib3.exceptions import ResponseError -import torch.nn.functional as F import tritonclient.grpc as grpcclient -import io +from urllib3.exceptions import ResponseError from app.core.config import * from app.service.design.utils.design_ensemble import get_keypoint_result @@ -179,14 +175,14 @@ class DesignPreprocessing: image_bytes = cv2.imencode(".jpg", ret)[1].tobytes() # image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}" bucket_name = image['image_url'].split('/', 1)[0] - object_name = image['image_url'].split('/', 1)[1] + object_name = image['image_url'].split('/', 1)[1].replace('.', '-show.') oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) image['show_image_url'] = f"{bucket_name}/{object_name}" else: image_bytes = cv2.imencode(".jpg", image['obj'])[1].tobytes() # image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}" bucket_name = image['image_url'].split('/', 1)[0] - object_name = image['image_url'].split('/', 1)[1] + object_name = image['image_url'].split('/', 1)[1].replace('.', '-show.') oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) image['show_image_url'] = f"{bucket_name}/{object_name}"