From 78c9420e09a5efc9a8225dcb9cfe6a250f4018f0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 13 Jul 2020 13:27:46 -0700 Subject: [PATCH] --classes bug fix #387 --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index a6b1f13..5c9577b 100644 --- a/detect.py +++ b/detect.py @@ -145,7 +145,7 @@ if __name__ == '__main__': parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') parser.add_argument('--view-img', action='store_true', help='display results') parser.add_argument('--save-txt', action='store_true', help='save results to *.txt') - parser.add_argument('--class', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3') + parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3') parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS') parser.add_argument('--augment', action='store_true', help='augmented inference') parser.add_argument('--update', action='store_true', help='update all models')