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

Update evaluation.py

Browse Source
This commit is contained in:
nihui 2022-07-07 21:01:05 +08:00 committed by GitHub
parent 559a0e95b9
commit d9f65e3755
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/evaluation.py
View File

@ -68,7 +68,7 @@ class CocoDetectionEvaluator():
# 模型预测 # 模型预测
preds = model(imgs) preds = model(imgs)
# 特征图后处理 # 特征图后处理
output = handel_preds(preds, self.device, 0.001) output = handle_preds(preds, self.device, 0.001)
# 检测结果 # 检测结果
N, _, H, W = imgs.shape N, _, H, W = imgs.shape