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

feat

Browse Source 
fix design 预处理seg结果逻辑修复
This commit is contained in:
zchen
2024-08-09 15:42:43 +08:00
parent 281f812636
commit ef78528e47
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/design_pre_processing/service.py
View File

@@ -151,10 +151,10 @@ class DesignPreprocessing:
# 推理得到keypoint # 推理得到keypoint
sketch['keypoint_result'] = self.keypoint_cache(sketch) sketch['keypoint_result'] = self.keypoint_cache(sketch)
if sketch['site'] == 'up': if sketch['site'] == 'up':
_, seg_cache = self.load_seg_result(sketch['image_id']) _, seg_cache = self.load_seg_result(sketch['obj'])
if not _: if not _:
# 推理获得seg 结果 # 推理获得seg 结果
seg_result = get_seg_result(sketch["image_id"], sketch['image_obj'])[0] seg_result = get_seg_result(sketch["image_id"], sketch['obj'])[0]
self.save_seg_result(seg_result, sketch['image_id']) self.save_seg_result(seg_result, sketch['image_id'])
if IF_DEBUG_SHOW: if IF_DEBUG_SHOW: