glenn-jocher
commited on
Commit
•
094079b
1
Parent(s):
1b1681b
update detect.py --class argument #387
Browse files
detect.py
CHANGED
@@ -145,7 +145,7 @@ if __name__ == '__main__':
|
|
145 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
146 |
parser.add_argument('--view-img', action='store_true', help='display results')
|
147 |
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
|
148 |
-
parser.add_argument('--
|
149 |
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
|
150 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
151 |
parser.add_argument('--update', action='store_true', help='update all models')
|
|
|
145 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
146 |
parser.add_argument('--view-img', action='store_true', help='display results')
|
147 |
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
|
148 |
+
parser.add_argument('--class', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3')
|
149 |
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
|
150 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
151 |
parser.add_argument('--update', action='store_true', help='update all models')
|