From 559a0e95b9d19e6332b6b1a07342837f7c3ee8b0 Mon Sep 17 00:00:00 2001 From: nihui Date: Thu, 7 Jul 2022 21:00:42 +0800 Subject: [PATCH] Update tool.py --- utils/tool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/tool.py b/utils/tool.py index 021b71d..fd4ee4f 100644 --- 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 @@ -117,4 +117,4 @@ def handel_preds(preds, device, conf_thresh=0.25, nms_thresh=0.45): for i in keep: output.append(temp[i]) output_bboxes.append(torch.Tensor(output)) - return output_bboxes \ No newline at end of file + return output_bboxes