From 67f52ba73c0be6696768fdfca75790ea69a8d1d0 Mon Sep 17 00:00:00 2001 From: Guo Haria Date: Sun, 27 Mar 2022 15:19:38 +0800 Subject: [PATCH] Update yolov5.py (#3656) fix rect position bug. --- python/ncnn/model_zoo/yolov5.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ncnn/model_zoo/yolov5.py b/python/ncnn/model_zoo/yolov5.py index c564c1ae5..d70592fdd 100644 --- a/python/ncnn/model_zoo/yolov5.py +++ b/python/ncnn/model_zoo/yolov5.py @@ -261,8 +261,8 @@ class YoloV5s: 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, )