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

feat

Browse Source 
fix     seg 新增preview_submit判断 当preview时不保存seg缓存
This commit is contained in:
zhouchengrong
2024-10-08 15:41:07 +08:00
parent df5e01ea06
commit e1677feb02
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/service/design_fast/pipeline/segmentation.py
View File

@@ -32,15 +32,21 @@ class Segmentation:
result['front_mask'] = np.array(red_mask, dtype=np.uint8) * 255 result['front_mask'] = np.array(red_mask, dtype=np.uint8) * 255
result['back_mask'] = np.array(green_mask, dtype=np.uint8) * 255 result['back_mask'] = np.array(green_mask, dtype=np.uint8) * 255
result['mask'] = result['front_mask'] + result['back_mask'] result['mask'] = result['front_mask'] + result['back_mask']
else:
# design信号判断 preview 不保存seg缓存
if "preview_submit" in result.keys() and result['preview_submit'] == "preview":
# 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['image'])[0]
else: else:
# 本地查询seg 缓存是否存在 # 本地查询seg 缓存是否存在
_, seg_result = self.load_seg_result(result["image_id"]) _, seg_result = self.load_seg_result(result["image_id"])
result['seg_result'] = seg_result
# 判断缓存和实际图片size是否相同 # 判断缓存和实际图片size是否相同
if not _ or result["image"].shape[:2] != seg_result.shape: if not _ or result["image"].shape[:2] != seg_result.shape:
# 推理获得seg 结果 # 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['image'])[0] seg_result = get_seg_result(result["image_id"], result['image'])[0]
self.save_seg_result(seg_result, result['image_id']) self.save_seg_result(seg_result, result['image_id'])
result['seg_result'] = seg_result
# 处理前片后片 # 处理前片后片
temp_front = seg_result == 1.0 temp_front = seg_result == 1.0
result['front_mask'] = (255 * (temp_front + 0).astype(np.uint8)) result['front_mask'] = (255 * (temp_front + 0).astype(np.uint8))