diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index bd6f81c..1b4f33c 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -142,6 +142,7 @@ class DesignPreprocessing: # @ RunTime def infer_image(self, image_list): + seg_result = None for sketch in image_list: # 小写 image_category = sketch['image_category'].lower() @@ -166,6 +167,7 @@ class DesignPreprocessing: points_list.append((int(i[1]), int(i[0]))) for point in points_list: cv2.circle(debug_show_image, point, point_size, point_color, thickness) + cv2.imshow("seg_result", seg_result) cv2.imshow("", debug_show_image) cv2.waitKey(0) # # 关键点在上部则推理seg