Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-17 16:11:26 +02:00
parent 0b8a28e3dd
commit a95e47533a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test.py
View File

@ -69,8 +69,9 @@ def test(
inf_out, train_out = model(imgs) # inference and training outputs
# Compute loss
loss_i, _ = compute_loss(train_out, targets, model)
loss += loss_i.item()
if hasattr(model, 'hyp'): # if model has loss hyperparameters
loss_i, _ = compute_loss(train_out, targets, model)
loss += loss_i.item()
# Run NMS
output = non_max_suppression(inf_out, conf_thres=conf_thres, nms_thres=nms_thres)