search

ohhhyeahhh / SiamGAT

Code for the paper "Graph Attention Tracking". (CVPR2021)
https://openaccess.thecvf.com/content/CVPR2021/papers/Guo_Graph_Attention_Tracking_CVPR_2021_paper.pdf
139 stars 19 forks source link

只想用一个数据集got-10k训练，会报找不到其他数据集的json文件 #26

Open vousemevoyez1024 opened 2 years ago

vousemevoyez1024 commented 2 years ago

请问一下各位要怎么解决呢？改哪里的代码？深度学习刚入门，谢谢！

twotwo2 commented 2 years ago

训练代码train.py中默认使用的是'../experiments/siamgat_googlenet/config.yaml'，对应的是全数据集训练。如果只需要got10k的数据集训练，将该配置修改为'../experiments/siamgat_googlenet_got10k/config.yaml'。配置文件config.yaml中指定训练数据的部分为：

DATASET:
    NAMES:
    - 'GOT'