aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-06-19 10:53:11 +08:00
parent 61ae688dd6
commit 8476bb3727
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_design.py
View File

@@ -32,8 +32,10 @@ def get_progress(request_data: DesignProgressModel):
process_id = request_data.process_id process_id = request_data.process_id
r = Redis() r = Redis()
data = r.read(key=process_id) 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}") logging.info(f"get_progress process_id @@@@@@ : {process_id} , progress : {data}")
except Exception as e: 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)) raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data) return ResponseModel(data=data)

4
app/service/design/items/pipelines/keypoints.py
View File

@@ -34,9 +34,9 @@ class KeypointDetection(object):
site = 'up' if result['name'] in ['blouse', 'outwear', 'dress', 'tops'] else 'down' site = 'up' if result['name'] in ['blouse', 'outwear', 'dress', 'tops'] else 'down'
# keypoint_cache = search_keypoint_cache(result["image_id"], site) # 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: if keypoint_cache is False:
keypoint_infer_result, site = self.infer_keypoint_result(result) keypoint_infer_result, site = self.infer_keypoint_result(result)