From ec649152e3f3a48605e3037cb7b946b497a5c6f0 Mon Sep 17 00:00:00 2001 From: zcr Date: Fri, 23 Jan 2026 17:34:51 +0800 Subject: [PATCH] =?UTF-8?q?=E7=A7=BB=E9=99=A4keypoint=20=E7=BC=93=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_pre_processing/service.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index 4c55997..9c9dd66 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -7,7 +7,7 @@ import numpy as np import torch import tritonclient.grpc as grpcclient from minio import Minio -from pymilvus import MilvusClient +# from pymilvus import MilvusClient from urllib3.exceptions import ResponseError from app.core.config import settings, SR_MODEL_NAME, SR_TRITON_URL, MILVUS_TABLE_KEYPOINT, KEYPOINT_RESULT_TABLE_FIELD_SET @@ -276,14 +276,15 @@ class DesignPreprocessing: def keypoint_cache(self, sketch): try: - client = MilvusClient(uri=settings.MILVUS_URL, token=settings.MILVUS_TOKEN, db_name=settings.MILVUS_ALIAS) + # client = MilvusClient(uri=settings.MILVUS_URL, token=settings.MILVUS_TOKEN, db_name=settings.MILVUS_ALIAS) keypoint_id = sketch['image_id'] - res = client.query( - collection_name=MILVUS_TABLE_KEYPOINT, - # ids=[keypoint_id], - filter=f"keypoint_id == {keypoint_id}", - output_fields=['keypoint_vector', 'keypoint_site'] - ) + # res = client.query( + # collection_name=MILVUS_TABLE_KEYPOINT, + # # ids=[keypoint_id], + # filter=f"keypoint_id == {keypoint_id}", + # output_fields=['keypoint_vector', 'keypoint_site'] + # ) + res = [] if len(res) == 0: # 没有结果 直接推理拿结果 并保存 keypoint_infer_result = self.infer_keypoint_result(sketch)