From 49d47adf17419fd8b9df6e5eeeed5870f19779bd Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 16 Feb 2020 23:30:14 -0800 Subject: [PATCH] updates --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index ca6a3873..22f927c2 100644 --- a/detect.py +++ b/detect.py @@ -164,7 +164,7 @@ if __name__ == '__main__': parser.add_argument('--output', type=str, default='output', help='output folder') # output folder parser.add_argument('--img-size', type=int, default=416, help='inference size (pixels)') parser.add_argument('--conf-thres', type=float, default=0.3, help='object confidence threshold') - parser.add_argument('--iou-thres', type=float, default=0.5, help='IOU threshold for NMS') + parser.add_argument('--iou-thres', type=float, default=0.6, help='IOU threshold for NMS') parser.add_argument('--fourcc', type=str, default='mp4v', help='output video codec (verify ffmpeg support)') 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')