diff --git a/predict/predict_yolo11seg.py b/predict/predict_yolo11seg.py
index 71685b3..48dfe12 100644
--- a/predict/predict_yolo11seg.py
+++ b/predict/predict_yolo11seg.py
@@ -459,7 +459,7 @@ class YOLOSegmentationInference:
                             approx = cv2.approxPolyDP(contour, epsilon, True)
 
                             # 归一化坐标
-                            h, w = mask.shape
+                            h, w = result.original_image.shape[:2]
                             points = []
                             for point in approx:
                                 x = point[0][0] / w