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-09-26 14:35:37 +08:00
parent 8968e914c9
commit 404cb079a9
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/service/design_fast/pipeline/keypoint.py
View File

@@ -23,10 +23,9 @@ class KeyPoint:
# result['clothes_keypoint'] = self.infer_keypoint_result(result) # result['clothes_keypoint'] = self.infer_keypoint_result(result)
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 # 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)
result['clothes_keypoint'] = self.save_keypoint_cache(result["image_id"], keypoint_infer_result, site) result['clothes_keypoint'] = self.save_keypoint_cache(result["image_id"], keypoint_infer_result, site)