修复

Egrt · Feb 7, 2023 · a4aed57 · a4aed57
1 parent 0f81b03
commit a4aed57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
                 #------------------------------#