design design batch

app/service/design_batch/pipeline/keypoint.py

@@ -4,7 +4,8 @@ import numpy as np
 from pymilvus import MilvusClient
 
 from app.core.config import *
-from app.service.design_batch.utils.design_ensemble import get_keypoint_result
+from app.service.design_fast.utils.design_ensemble import get_keypoint_result
+from app.service.utils.decorator import ClassCallRunTime, RunTime
 
 logger = logging.getLogger(__name__)
 
@@ -16,14 +17,15 @@ class KeyPoint:
     def get_name(cls):
         return cls.name
 
+    @ClassCallRunTime
     def __call__(self, result):
         if result['name'] in ['blouse', 'skirt', 'dress', 'outwear', 'trousers', 'tops', 'bottoms']:  # 查询是否有数据 且类别相同 相同则直接读 不同则推理后更新
             # result['clothes_keypoint'] = self.infer_keypoint_result(result)
             site = 'up' if result['name'] in ['blouse', 'outwear', 'dress', 'tops'] else 'down'
             # keypoint_cache = search_keypoint_cache(result["image_id"], site)
-            keypoint_cache = self.keypoint_cache(result, site)
+            # keypoint_cache = self.keypoint_cache(result, site)
+            keypoint_cache = False
             # 取消向量查询 直接过模型推理
-            # keypoint_cache = False
             if keypoint_cache is False:
                 keypoint_infer_result, site = self.infer_keypoint_result(result)
                 result['clothes_keypoint'] = self.save_keypoint_cache(result["image_id"], keypoint_infer_result, site)
@@ -87,7 +89,7 @@ class KeyPoint:
             logger.info(f"save keypoint cache milvus error : {e}")
             return dict(zip(KEYPOINT_RESULT_TABLE_FIELD_SET, result.reshape(12, 2).astype(int).tolist()))
 
-    # @ RunTime
+    @RunTime
     def keypoint_cache(self, result, site):
         try:
             client = MilvusClient(uri=MILVUS_URL, token=MILVUS_TOKEN, db_name=MILVUS_ALIAS)