From 7f94270d83f4e3cfa87dbf6d027ace8bca2fa81b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E9=B9=AD=E5=85=88=E7=94=9F?= <766529835@qq.com>
Date: Thu, 2 Feb 2023 20:44:34 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=A7=A3=E8=80=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 yolo.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/yolo.py b/yolo.py
index ab053df..dce3a4d 100644
--- a/yolo.py
+++ b/yolo.py
@@ -161,8 +161,8 @@ class YOLO(object):
             #---------------------------------------------------------#
             #   将归一化的预测结果变为真实的预测框
             #---------------------------------------------------------#
-            top_polys[..., [0, 2, 4, 6]] *= image_shape[0]
-            top_polys[..., [1, 3, 5, 7]] *= image_shape[1]
+            top_polys[..., [0, 2, 4, 6]] *= image_shape[1]
+            top_polys[..., [1, 3, 5, 7]] *= image_shape[0]
         #---------------------------------------------------------#
         #   设置字体与边框厚度
         #---------------------------------------------------------#
-- 
GitLab