Update evaluation.py
This commit is contained in:
parent
559a0e95b9
commit
d9f65e3755
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user