From 8476bb3727e3ed974dad3f09d2ca3050e8e3da7e Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Wed, 19 Jun 2024 10:53:11 +0800 Subject: [PATCH] feat fix --- app/api/api_design.py | 4 +++- app/service/design/items/pipelines/keypoints.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/app/api/api_design.py b/app/api/api_design.py index ecac4f5..cdbd1f5 100644 --- a/app/api/api_design.py +++ b/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) diff --git a/app/service/design/items/pipelines/keypoints.py b/app/service/design/items/pipelines/keypoints.py index 956e052..6cf1141 100644 --- a/app/service/design/items/pipelines/keypoints.py +++ b/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)