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 @@ class EvalCallback(): #------------------------------# # 获得预测框 #------------------------------# - 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 #------------------------------# -- GitLab