Open sfarkya04 opened 2 years ago
Hi~
For multiclass object tracking, you should change models/tracker.py
The learning rate, batch size should be adjusted based on num_gpus. This codebase will automatically download Imagenet pre-trained model.
Thanks for replying @PeizeSun
I have made the changes you suggested. I am not clear about how it's automatically downloading the imagenet pre-trained weights.
I assumed that it must be happening here: tracking_code
But based on what you said I should not set resume param and the code will take care of downloading the imagenet default weights? My concern is that I didn't see any weights being downloaded and even if they are downloaded then not sure how are they loaded? Would be really helpful you could tell me that. Maybe it's a PyTorch thing I don't know about?
Thanks again for helping! :)
Hello! I'd like to confirm that if I want to train TransTrack for multi-object tracking, do I only need to modify tracker.py in the model code? Since the Tracker class is only used during inference, is it correct to say that I don't need to modify any training code except for the data loading part? Thank you in advance!
Hello Dear Authors, First of all, thank you for providing code and wonderful work.
I am trying to use your codebase to perform tracking on bdd100k dataset. https://github.com/bdd100k/bdd100k
The dataset has 8 classes: car(dominated), truck, bus, pedestrian etc. Originally, bdd100k has 1400 training videos which is huge so created a smaller set with 100 videos to train the model The devset has about 20000 training images and 8 classes.
I followed this and modified the build function: https://github.com/PeizeSun/TransTrack/issues/11
Also, you can use for the conversion: https://doc.bdd100k.com/format.html
I gave a dry run on my side to make sure the code is working for bdd to mot converted annotations and it seems to work fine.
I don't think I have to change the num_classes since you already have set it to 20?
Now I am confused by the comment made here: https://github.com/PeizeSun/TransTrack/issues/41#issuecomment-939762956 Essentially it says: "we should make sure the category is kept for the objects to be associated." Not sure where I have to make the change to incorporate that.
Further, I have some other questions to make sure the setup is correct: Resources I have; 4, 48 NVIDIA GPUs to train the model.