diff --git a/utils/callbacks.py b/utils/callbacks.py index a378ac2c99c020164cf641c10dd3c0ec048d4e9d..a703585e5c101318bb2d35206cccfab208dc6ae5 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 #------------------------------#