yaclty/FastestDet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tool.py

Browse Source
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
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/tool.py
View File

@ -58,7 +58,7 @@ class EMA():
self.backup = {} 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 = [], [] total_bboxes, output_bboxes = [], []
# 将特征图转换为检测框的坐标 # 将特征图转换为检测框的坐标
N, C, H, W = preds.shape 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: for i in keep:
output.append(temp[i]) output.append(temp[i])
output_bboxes.append(torch.Tensor(output)) output_bboxes.append(torch.Tensor(output))
return output_bboxes return output_bboxes