From 3d6210adb853697851ae28cb21cee3d420b8b732 Mon Sep 17 00:00:00 2001
From: yooooger <761181201@qq.com>
Date: Thu, 5 Mar 2026 14:18:49 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BD=92=E4=B8=80=E5=8C=96=E5=9D=90=E6=A0=87?=
 =?UTF-8?q?=E6=94=B9=E7=94=A8result.original=5Fimage.shape[:2]?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 predict/predict_yolo11seg.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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