Training on Custom Dataset

[sakshi-2412]

Hey, your work looks impressive!
I wanted to fine-tune SMAT on a custom dataset. However, changing all the configurations and files seems a little complex. Is there any chance that training on a custom dataset feature is also added in the codebase? Thanks!

[goutamyg]

Thank you for your interest in our work!

The task of preparing multiple datasets for model training is implemented here. We have borrowed it from the previously published tracker implementations. We haven't used a new dataset other than LaSOT + GOT10k + TrackingNet + MS-COCO for model training; however, you can follow these scripts to integrate a image and video-based custom datasets to fine-tune the model.

If you want to set varying learning rates for different components in the network, you can refer to this function.