From 23288236a6192736d1e98548705e1f98a2d64f02 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 2 Jan 2020 09:50:11 -0800 Subject: [PATCH] updates --- detect.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index da919b37..b9ddbf0f 100644 --- a/detect.py +++ b/detect.py @@ -6,9 +6,9 @@ from utils.datasets import * from utils.utils import * -def detect(save_txt=False, save_img=False): +def detect(save_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, view_img = opt.output, opt.source, opt.weights, opt.half, opt.view_img + out, source, weights, half, view_img, save_txt = opt.output, opt.source, opt.weights, opt.half, opt.view_img, opt.save_txt webcam = source == '0' or source.startswith('rtsp') or source.startswith('http') or source.endswith('.txt') # Initialize @@ -167,6 +167,7 @@ if __name__ == '__main__': parser.add_argument('--half', action='store_true', help='half precision FP16 inference') parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1) or cpu') parser.add_argument('--view-img', action='store_true', help='display results') + parser.add_argument('--save-txt', action='store_true', help='display results') parser.add_argument('--classes', nargs='+', type=int, help='filter by class') opt = parser.parse_args() print(opt)