From a4aed57fac1b6d2ed41a10bed88c9b0f7d8ae5c0 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: Tue, 7 Feb 2023 20:56:42 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/callbacks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/callbacks.py b/utils/callbacks.py
index a378ac2..a703585 100644
--- a/utils/callbacks.py
+++ b/utils/callbacks.py
@@ -195,7 +195,7 @@ def on_epoch_end(self, epoch, model_eval):
                 #------------------------------#
                 #   获得预测框
                 #------------------------------#
-                gt_boxes    = np.array([np.array(list(map(int,box.split(',')))) for box in line[1:]])
+                gt_boxes    = np.array([np.array(list(map(float,box.split(',')))) for box in line[1:]])
                 #------------------------------#
                 #   将polygon转换为hbb
                 #------------------------------#