feat sketch 提取接口

fix
2024-08-14 16:45:34 +08:00
parent 281f812636
commit d11beb0107
25 changed files with 2681 additions and 8 deletions
--- 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: