Update yolov5.py (#3656)

fix rect position bug.
Tencent · Mar 27, 2022 · 67f52ba · 67f52ba
1 parent 94beeaf
commit 67f52ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ncnn/model_zoo/yolov5.py b/python/ncnn/model_zoo/yolov5.py
@@ -261,8 +261,8 @@ def __call__(self, img):
             Detect_Object(
                 obj[5],
                 obj[4],
-                obj[0] / scale,
-                obj[1] / scale,
+                (obj[0] - (wpad / 2)) / scale,
+                (obj[1] - (hpad / 2)) / scale,
                 (obj[2] - obj[0]) / scale,
                 (obj[3] - obj[1]) / scale,
             )