From aa501d848aabd6c99edc38dc41a32eddc1d0dea0 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 19 Aug 2024 11:10:48 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20=20sketch=20=E6=8F=90=E5=8F=96?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3=20=E5=92=8C=20minio=E8=B6=85=E6=97=B6?= =?UTF-8?q?=E8=AE=BE=E7=BD=AE=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design_pre_processing/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index 4cbff8f..70502d0 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -151,7 +151,7 @@ class DesignPreprocessing: # 推理得到keypoint sketch['keypoint_result'] = self.keypoint_cache(sketch) if sketch['site'] == 'up': - _, seg_cache = self.load_seg_result(sketch['obj']) + _, seg_cache = self.load_seg_result(sketch['image_id']) if not _: # 推理获得seg 结果 seg_result = get_seg_result(sketch["image_id"], sketch['obj'])[0]