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

updates

This commit is contained in:
Glenn Jocher 2019-02-11 14:19:06 +01:00
parent 0d1bcae1ef
commit dea82efa29
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
detect.py
View File

@ -18,7 +18,7 @@ def detect(
nms_thres=0.45, nms_thres=0.45,
save_txt=False, save_txt=False,
save_images=True, save_images=True,
webcam=False webcam=True
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
os.system('rm -rf ' + output) os.system('rm -rf ' + output)
@ -85,13 +85,14 @@ def detect(
label = '%s %.2f' % (classes[int(cls)], conf) label = '%s %.2f' % (classes[int(cls)], conf)
plot_one_box([x1, y1, x2, y2], im0, label=label, color=colors[int(cls)]) plot_one_box([x1, y1, x2, y2], im0, label=label, color=colors[int(cls)])
print('Done. (%.3fs)' % (time.time() - t)) dt = time.time() - t
print('Done. (%.3fs)' % dt)
if save_images: # Save generated image with detections if save_images: # Save generated image with detections
cv2.imwrite(save_path, im0) cv2.imwrite(save_path, im0)
if webcam: # Show live webcam if webcam: # Show live webcam
cv2.imshow(weights, im0) cv2.imshow(weights + ' - %.2f FPS' % (1 / dt), im0)
if save_images and (platform == 'darwin'): # MacOS if save_images and (platform == 'darwin'): # MacOS
os.system('open ' + output + '&& open ' + save_path) os.system('open ' + output + '&& open ' + save_path)