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

updates

This commit is contained in:
Glenn Jocher 2019-09-27 23:46:45 +02:00
parent 4aa60ea499
commit b694e52e2d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
detect.py
View File

@ -69,6 +69,9 @@ def detect(save_txt=False, save_img=False):
img = img.unsqueeze(0) img = img.unsqueeze(0)
pred, _ = model(img) pred, _ = model(img)
if opt.half:
pred = pred.float()
for i, det in enumerate(non_max_suppression(pred, opt.conf_thres, opt.nms_thres)): # detections per image for i, det in enumerate(non_max_suppression(pred, opt.conf_thres, opt.nms_thres)): # detections per image
if webcam: # batch_size >= 1 if webcam: # batch_size >= 1
p, s, im0 = path[i], '%g: ' % i, im0s[i] p, s, im0 = path[i], '%g: ' % i, im0s[i]