diff --git a/test.py b/test.py index 4f5671f..b43bc83 100644 --- a/test.py +++ b/test.py @@ -69,7 +69,7 @@ if __name__ == '__main__': print("forward time:%fms"%time) # 特征图后处理 - output = handel_preds(preds, device, opt.thresh) + output = handle_preds(preds, device, opt.thresh) # 加载label names LABEL_NAMES = [] diff --git a/utils/evaluation.py b/utils/evaluation.py index f00f5cc..f7123b2 100644 --- a/utils/evaluation.py +++ b/utils/evaluation.py @@ -68,7 +68,7 @@ class CocoDetectionEvaluator(): # 模型预测 preds = model(imgs) # 特征图后处理 - output = handel_preds(preds, self.device, 0.001) + output = handle_preds(preds, self.device, 0.001) # 检测结果 N, _, H, W = imgs.shape 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