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

updates

This commit is contained in:
Glenn Jocher 2019-08-01 02:28:11 +02:00
parent e3d100dd34
commit f9d616de9f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -77,7 +77,7 @@ def detect(cfg,
# Get detections # Get detections
img = torch.from_numpy(img).unsqueeze(0).to(device) img = torch.from_numpy(img).unsqueeze(0).to(device)
pred, _ = model(img) pred, _ = model(img)
det = non_max_suppression(pred, conf_thres, nms_thres)[0] det = non_max_suppression(pred.float(), conf_thres, nms_thres)[0]
if det is not None and len(det) > 0: if det is not None and len(det) > 0:
# Rescale boxes from 416 to true image size # Rescale boxes from 416 to true image size