apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher
2020-01-08 09:42:01 -08:00
parent bf42c31d9e
commit 11ce877bdf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+1 -1
View File
@@ -540,7 +540,7 @@ def non_max_suppression(prediction, conf_thres=0.5, iou_thres=0.5, multi_cls=Tru
# Filter by class # Filter by class
if classes: if classes:
pred = pred[(j.view(-1, 1) == torch.Tensor(classes)).any(1)] pred = pred[(j.view(-1, 1) == torch.tensor(classes, device=j.device)).any(1)]
# Apply finite constraint # Apply finite constraint
if not torch.isfinite(pred).all(): if not torch.isfinite(pred).all():