pytracking-ATOM代码--用自己训练的模型进行跟踪

  1. 把ltr.result_model.checkpoints.ltr.bbreg.atom下的文件ATOMnet_ep0050pth.tar移动到pytracking.networks下,并改名去掉.tar后缀(即解压);
    在这里插入图片描述
    pytracking-ATOM代码--用自己训练的模型进行跟踪_第1张图片

  2. 到pytracking.parameter.atom.default.py文件的最后,改写deep_feat那句,把net_path改成刚刚移动的那个文件

deep_feat = deep.ATOMResNet18(net_path='atom_default.pth', output_layers=['layer3'], fparams=deep_fparams, normalize_power=2)
 deep_feat = deep.ATOMResNet18(net_path='ATOMnet_ep0050.pth', output_layers=['layer3'], fparams=deep_fparams,normalize_power=2)

你可能感兴趣的:(#,pytorch,#,目标跟踪,笔记,深度学习,pytorch,神经网络)