pysot系列--test.py笔记（加载数据集获取跟踪算法结果并保存）

# Copyright (c) SenseTime. All Rights Reserved.

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import sys
#sys.path.append('你的\\pysot-master\\路径')
sys.path.append('C:\\Users\\lpf\\Desktop\\Code_of_Paper\\pysot-OTB2015\\pysot')
sys.path.append('C:\\Users\\lpf\\Desktop\\Code_of_Paper\\pysot-OTB2015')
import argparse
import os

import cv2
import torch
import numpy as np

from pysot.core.config import cfg
from pysot.models.model_builder import ModelBuilder
from pysot.tracker.tracker_builder import build_tracker
from pysot.utils.bbox import get_axis_aligned_bbox
from pysot.utils.model_load import load_pretrain
from toolkit.datasets import DatasetFactory
from toolkit.utils.region import vot_overlap, vot_float2str


parser = argparse.ArgumentParser(description='siamrpn tracking')

dataset_name = 'OTB100'
# snapshot = 'C:/Users/lpf/Desktop/Code_of_Paper/PySOT/siamrpn_alex_dwxcorr/model.pth'
snapshot = 'C:/Users/lpf/Desktop/Code_of_Paper/PySOT/siamrpn_r50_l234_dwxcorr/model.pth'
# config = r'../experiments/siamrpn_alex_dwxcorr/config.yaml'
config = r'../experiments/siamrpn_r50_l234_dwxcorr/config.yaml'

parser.add_argument('--dataset', default=dataset_name, type=str, help='datasets')#数据集
parser.add_argument('--config', default=config, type=str, help='config file')#配置文件
parser.add_argument('--snapshot', default=snapshot, type=str,help='snapshot of models to eval')#model path
parser.add_argument('--video', default='', type=str, help='eval one special video')
parser.add_argument('--vis', action='store_true', help='whether visualzie result')#可视化数据

args = parser.parse_args()

torch.set_num_threads(8)

dataset_root = r"C:\Users\lpf\Desktop\Code_of_Paper\pysot-OTB2015\testing_dataset\OTB100"

def main():
    # load config
    cfg.merge_from_file(args.config)

    #os.path.realpath: 获取的__file__所在脚本的路径，也就是fileName.py的路径
    #os.path.dirname 功能：去掉文件名，返回目录
    #cur_dir = C:\Users\lpf\Desktop\Code_of_Paper\pysot-master\tools\
    # cur_dir = os.path.dirname(os.path.realpath(__file__))

    # dataset_root = os.path.join(cur_dir, '../testing_dataset', args.dataset)
    # root = os.path.realpath(os.path.join(cur_dir, '../testing_dataset'))
    # dataset_root = os.path.join(root, args.dataset)
    # root = r'C:\Users\lpf\Desktop\dataset\VOT'
    # root = r'C:\Users\lpf\Desktop\Code_of_Paper\pysot-master\tools'
    # dataset_root = os.path.join(root, args.dataset)

    # create model
    model = ModelBuilder()

    # load model
    model = load_pretrain(model, args.snapshot).cuda().eval()

    # build tracker
    tracker = build_tracker(model)

    # create dataset
    dataset = DatasetFactory.create_dataset(name=args.dataset,
                                            dataset_root=dataset_root,
                                            load_img=False)

    model_name = args.snapshot.split('/')[-1].split('.')[0]
    total_lost = 0
    # VOT16 17 18 19 短时视频跟踪结果
    if args.dataset in ['VOT2016', 'VOT2018', 'VOT2019']:
        # restart tracking
        for v_idx, video in enumerate(dataset):
            if args.video != '':
                # test one special video