归一化坐标改用result.original_image.shape[:2]

2026-03-05 14:18:49 +08:00 · 2026-03-05 14:18:49 +08:00 · 3d6210adb8
commit 3d6210adb8
parent d4151c2dab
1 changed files with 1 additions and 1 deletions
--- 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