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 #------------------------------#