Update tool.py

2022-07-07 21:00:42 +08:00 · 2022-07-07 21:00:42 +08:00 · 559a0e95b9
commit 559a0e95b9
parent 8d49c6a969
1 changed files with 2 additions and 2 deletions
--- a/utils/tool.py
+++ b/utils/tool.py
@ -58,7 +58,7 @@ class EMA():
        self.backup = {}

 # 后处理(归一化后的坐标)
-def handel_preds(preds, device, conf_thresh=0.25, nms_thresh=0.45):
+def handle_preds(preds, device, conf_thresh=0.25, nms_thresh=0.45):
    total_bboxes, output_bboxes  = [], []
    # 将特征图转换为检测框的坐标
    N, C, H, W = preds.shape