Update tool.py

This commit is contained in:
nihui 2022-07-07 21:00:42 +08:00 committed by GitHub
parent 8d49c6a969
commit 559a0e95b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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
return output_bboxes