ATOM代码运行记录（测试加训练）

ATOM代码运行记录（测试加训练）

pytracking系列跟踪的环境配置见博客

代码测试

#在pytracking/run_tracker.py下有运行参数
parser = argparse.ArgumentParser(description='Run tracker on sequence or dataset.')
#需要运行的跟踪器名称，在pytracking/tracker/路径下
parser.add_argument('tracker_name', type=str, default='dimp', help='Name of tracking method.')
#需要运行的跟踪器参数，在pytracking/parameter/路径下每个跟踪器有多种参数
parser.add_argument('tracker_param', type=str, default='dimp50',help='Name of parameter file.')
parser.add_argument('--runid', type=int, default=None, help='The run id.')
#需要运行的数据集名称
parser.add_argument('--dataset_name', type=str, default='otb', help='Name of dataset (otb, nfs, uav, tpl, vot, tn, gott, gotv, lasot).')
#需要运行的序列名称
parser.add_argument('--sequence', type=str, default=None, help='Sequence number or name.')
#debug控制可视化等级，0表示提供最少的信息，9表示提供最多的信息，数越高,越详细；当debug为0时，只保存结果，没有可视化，当把 debug 等级设为8时，没有保存结果
parser.add_argument('--debug', type=int, default=0, help='Debug level.')
#threads运行的线程数
parser.add_argument('--threads', type=int, default=0, help='Number of threads.')
#使用可视化
parser.add_argument('--use_visdom', type=bool, default=True, help='Flag to enable visdom.')
#可视化服务器
parser.add_argument('--visdom_server', type=str, default='127.0.0.1', help='Server for visdom.')
#可视化端口
parser.add_argument('--visdom_port', type=int, default=8097, help='Port for visdom.')

测试某个序列

python run_tracker.py atom default --dataset otb --sequence Soccer --debug 0 --threads 8

测试某个数据集

python run_tracker.py atom default --dataset otb --debug 0 --threads 8

训练数据集

修改pytracking/ltr/admin/local.py,将需要训练的数据集路径添加上
在ltr/train_settings/bbreg/atom.py中第25行，选择要训练的数据集
第67行左右[lasot_train, got10k_train]训练集名称，比例[1，1]

dataset_train = sampler.ATOMSampler([lasot_train, got10k_train], [1,1],samples_per_epoch=1000*settings.batch_size, max_gap=50, processing=data_processing_train)

训练 ltr/train_settings/bbreg/atom.py

#bbreg为ltr/train_settings/bbreg默认atom文件名，atom为ltr/train_settings/bbreg/路径下训练参数文件
python run_training.py bbreg atom

tensorboard可视化训练过程

在训练好的路径下result_model\下，进入训练的tensorboard目录下

打开一个终端，激活当前环境，打开tensorboard

tensorboard --logdir=/home/bxy/pytracking/ltr/result_model/tensorboard/ltr/bbreg/atom/train

打开弹出的本地服务器网址http://localhost:6006/

参考博客

【1】目标跟踪之Pytracking系列代码训练(ATOM\DiMP\PrDiMP\KYS\Super_DiMP\KeepTrack)
【2】Pytracking代码运行笔记
【3】Tensorflow的可视化工具Tensorboard的初步使用
【4】pytracking-ATOM代码–用自己训练的模型进行跟踪
【5】ATOM网络模型