Update evaluation.py

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

View File

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