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-12 12:52:59 +02:00
parent 780fa17f6a
commit 121da9a6c0
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

@ -10,7 +10,7 @@ def detect(save_txt=False, save_img=False, stream_img=False):
img_size = (320, 192) if ONNX_EXPORT else opt.img_size # (320, 192) or (416, 256) or (608, 352) for (height, width)
out, source, weights, half = opt.output, opt.source, opt.weights, opt.half
webcam = source == '0' or source.startswith('rtsp') or source.startswith('http')
streams = source == 'streams.txt'
streams = source.endswith('streams.txt')
# Initialize
device = torch_utils.select_device(force_cpu=ONNX_EXPORT)