From b4490ebb951b0d3254f30fc3afc9f19918c4f4c4 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 24 Jun 2024 15:00:04 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20design=E9=A2=84=E5=A4=84?= =?UTF-8?q?=E7=90=86=20=E8=A1=A5=E5=81=BF=E7=99=BD=E8=BE=B9=E5=9B=BEurl?= =?UTF-8?q?=E9=81=97=E6=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_pre_processing/service.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) 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}"