PointTrack icon indicating copy to clipboard operation
PointTrack copied to clipboard

Pedestrain training

Open iamjeff7 opened this issue 5 years ago • 3 comments

Dear Author, How can I train for pedestrian instead of car. Thank you!

iamjeff7 avatar Jul 21 '20 04:07 iamjeff7

Similar question. @detectRecog will you release the pretrained weights for pedestrian

qa276390 avatar Sep 30 '20 07:09 qa276390

For segmentation, you can train on pedestrians by switching class_id to 2 rather than 1 (for cars) in the dataset. The change of code for tracking should also be similar. There are bugs in the pedestrian model. So we do not plan to provide our trained model for pedestrians. However, for the convenience of comparisons with PointTrack, we have provided the test set results of PointTrack on cars and pedestrians on the GitHub page.

detectRecog avatar Oct 18 '20 14:10 detectRecog

Hi, I want to train on pedestrian too, can u explain more specific? I don't find where to switch class_id, thank you very much! @detectRecog

EchoAmor avatar Nov 04 '20 02:11 EchoAmor