How to use my training model(.pth.tar file) to track with custom datasets ?

visionml / pytracking

Visual tracking library based on PyTorch.

GNU General Public License v3.0

3.19k stars 603 forks source link

How to use my training model(.pth.tar file) to track with custom datasets ? #418

Closed JiaoMX-keeping closed 7 months ago

JiaoMX-keeping commented 7 months ago

Hi, I appreciate your public release of this tracking package. I am very interested in this trackers and try to make meaningful experience with it.

I tried to train the ATOM tracker with my own dataset and got the pth file as shown in Fig. I want to use this file as a parameter for the tracking test, but the execution can not find the corresponding parameter. Do I need to write a parresolution file for a parameter file? What should I do, please?

Thanks very much !!!

JiaoMX-keeping commented 7 months ago

ok. u need to modify the file in pytracking/parameter/atom/xxxx.py.