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]