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: