timmeinhardt
Smaller Backbone Models
Hi Tim Meinhardt thanks for the great work!
I'm trying to adapt trackformer in my own research, however I don't have enough GPU memory. Do you have private mode models with smaller backbones(resnet18/34) pretrained on coco/crowdedhuman detection task or scripts for coco detection pretraining? Thanks!
I am sorry but we never trained smaller models than the provided ResNet50.
Okay. Do you have instructions/training scripts for coco pretraining then? I can also try to train it with smaller models myself. Thanks!
After downloading and unpacking the COCO dataset into the data directory you should be able to train a model on COCO detection by running something like this:
python src/train.py with \
deformable \
backbone=resnet18 \
output_dir=models/coco_resnet18
But this is not tested so make yourself familiar with the codebase to understand the configuration parameters in cfgs/train.yaml.