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

updates

This commit is contained in:
Glenn Jocher 2019-12-25 14:55:11 -08:00
parent 34d9392bac
commit d1087f4987
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/utils.py
View File

@ -537,7 +537,8 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5, multi_cls=Tru
pred = torch.cat((box, conf.unsqueeze(1), j.float().unsqueeze(1)), 1) pred = torch.cat((box, conf.unsqueeze(1), j.float().unsqueeze(1)), 1)
# Apply finite constraint # Apply finite constraint
pred = pred[torch.isfinite(pred).all(1)] if not torch.isfinite(pred).all():
pred = pred[torch.isfinite(pred).all(1)]
# Batched NMS # Batched NMS
if method == 'vision_batch': if method == 'vision_batch':