How to add Bicycle etc. clase?

JunweiLiang / Object_Detection_Tracking

Out-of-the-box code and models for CMU's object detection and tracking system for multi-camera surveillance videos. Speed optimized Faster-RCNN model. Tensorflow based. Also supports EfficientDet. WACVW'20

MIT License

479 stars 135 forks source link

How to add Bicycle etc. clase? #25

Closed isaryo closed 3 years ago

isaryo commented 3 years ago

This model is tracking "Person" and "Vehicle" class. How to add Bicycle etc. clase? Also if I want to add other classes, which code should I change?

JunweiLiang commented 3 years ago

See the --tracking_objs parameter here. The default is Person,Vehicle. You can simply add additional object class names (comma separated) according to the name list here. You can only do this with VIRAT trained object models (v1-6).