search

visionml / pytracking

Visual tracking library based on PyTorch.
GNU General Public License v3.0
3.22k stars 605 forks source link

Train on custom dataset #410

Closed setarekhosravi closed 4 months ago

setarekhosravi commented 10 months ago

Hi, thank you for your great work. I want to train ToMP on my own dataset, how can I prepare my data? thank you.

2006pmach commented 4 months ago

See here for how LaSOT is integrated. Simply create a new dataset class for your own dataset and create the sequences as explained in lasot.py. Then specify your dataset during training in the training config files.