From 5333bb6d6932ce2a1d1656b3fd324858e998a203 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Tue, 8 Oct 2024 16:01:08 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20=20=20=20seg=20=E6=96=B0?= =?UTF-8?q?=E5=A2=9Epreview=5Fsubmit=E5=88=A4=E6=96=AD=20=E5=BD=93preview?= =?UTF-8?q?=E6=97=B6=E4=B8=8D=E4=BF=9D=E5=AD=98seg=E7=BC=93=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_fast/pipeline/segmentation.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/app/service/design_fast/pipeline/segmentation.py b/app/service/design_fast/pipeline/segmentation.py index 786cf03..ebf02b4 100644 --- a/app/service/design_fast/pipeline/segmentation.py +++ b/app/service/design_fast/pipeline/segmentation.py @@ -33,10 +33,16 @@ class Segmentation: result['back_mask'] = np.array(green_mask, dtype=np.uint8) * 255 result['mask'] = result['front_mask'] + result['back_mask'] else: - # design信号判断 preview 不保存seg缓存 + # preview 过模型 不缓存 if "preview_submit" in result.keys() and result['preview_submit'] == "preview": # 推理获得seg 结果 seg_result = get_seg_result(result["image_id"], result['image'])[0] + # submit 过模型 缓存 + elif "preview_submit" in result.keys() and result['preview_submit'] == "submit": + # 推理获得seg 结果 + seg_result = get_seg_result(result["image_id"], result['image'])[0] + self.save_seg_result(seg_result, result['image_id']) + # null 正常流程 加载本地缓存 无缓存则过模型 else: # 本地查询seg 缓存是否存在 _, seg_result = self.load_seg_result(result["image_id"])