feat

fix

app/api/api_design.py

@@ -32,8 +32,10 @@ def get_progress(request_data: DesignProgressModel):
         process_id = request_data.process_id
         r = Redis()
         data = r.read(key=process_id)
+        if data is None:
+            raise ValueError("The progress must be numbers ")
         logging.info(f"get_progress process_id @@@@@@ : {process_id} , progress : {data}")
     except Exception as e:
-        logger.warning(f"design Run Exception @@@@@@:{e}")
+        logger.warning(f"get_progress Run Exception @@@@@@:{e}")
         raise HTTPException(status_code=404, detail=str(e))
     return ResponseModel(data=data)

app/service/design/items/pipelines/keypoints.py

@@ -34,9 +34,9 @@ class KeypointDetection(object):
             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)