From ef78528e4716764112fa348a487df08d7908aac8 Mon Sep 17 00:00:00 2001 From: zchen Date: Fri, 9 Aug 2024 15:42:43 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20design=20=E9=A2=84=E5=A4=84?= =?UTF-8?q?=E7=90=86seg=E7=BB=93=E6=9E=9C=E9=80=BB=E8=BE=91=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_pre_processing/service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index 1b4f33c..4cbff8f 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -151,10 +151,10 @@ class DesignPreprocessing: # 推理得到keypoint sketch['keypoint_result'] = self.keypoint_cache(sketch) if sketch['site'] == 'up': - _, seg_cache = self.load_seg_result(sketch['image_id']) + _, seg_cache = self.load_seg_result(sketch['obj']) if not _: # 推理获得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']) if IF_DEBUG_SHOW: