NauchtanRobotics
commited on
Commit
•
dad8660
1
Parent(s):
59aae85
Fix arg help string to match 'classes' arg name (#4893)
Browse files
detect.py
CHANGED
@@ -267,7 +267,7 @@ def parse_opt():
|
|
267 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|
268 |
parser.add_argument('--save-crop', action='store_true', help='save cropped prediction boxes')
|
269 |
parser.add_argument('--nosave', action='store_true', help='do not save images/videos')
|
270 |
-
parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --
|
271 |
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
|
272 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
273 |
parser.add_argument('--visualize', action='store_true', help='visualize features')
|
|
|
267 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|
268 |
parser.add_argument('--save-crop', action='store_true', help='save cropped prediction boxes')
|
269 |
parser.add_argument('--nosave', action='store_true', help='do not save images/videos')
|
270 |
+
parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --classes 0, or --classes 0 2 3')
|
271 |
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
|
272 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
273 |
parser.add_argument('--visualize', action='store_true', help='visualize features')
|