pytracking 部署笔记

目录

参数：

import sys sys.path.insert(0, r"E:/project/track/pytracking-master") from pytracking.evaluation import Tracker from pytracking.evaluation.data import Sequence, BaseDataset, SequenceList from pytracking.evaluation.running import run_sequence import os import numpy as np # 1. 定义你的数据集类 class MyCustomDataset(BaseDataset): def __init__(self): super().__init__() img_dir = r"C:\Users\ChanJing-01\Videos\yumao\f_400" gt_path = r"C:\Users\ChanJing-01\Videos\yumao\f_400\gt.txt" self.name = "my_own_dataset" self.base_path = r"E:/my_data/my_seq" self.img_dir = img_dir self.gt_path = gt_path # 读取图片列表 self.img_list = sorted([os.path.join(self.img_dir, f) for f in os.listdir(self.img_dir) if f.endswith(('.jpg', '.png'))]) # 读取gt self.gt_rect = np.loadtxt(self.gt_path, delimiter=',', dtype=np.float32) if self.gt_rect.ndim == 1: self.gt_rect = self.gt_rect.reshape(1, -1) # 准备序列列表 self.sequence_list = self._create_sequence_list() def _create_sequence_list(self): seq_list = SequenceList() seq = Sequence(name="my_sequence", frames=self.img_list, dataset=self, # 传入自己作为dataset ground_truth_rect=self.gt_rect, object_class="unknown", target_visible=None) seq_list.append(seq) return seq_list def __len__(self): return len(self.sequence_list) def get_sequence_list(self): return self.sequence_list # 2. 使用 dataset = MyCustomDataset() seq_list = dataset.get_sequence_list() seq = seq_list[0] # 获取第一个序列 # 3. 创建跟踪器 tracker = Tracker("dimp", "prdimp50") # 4. 运行 run_sequence(seq, tracker, debug=0)

参数：

E:\project\track\pytracking-master\pytracking\parameter\dimp\dimp18.py

本质是找代码路径：

parser.add_argument('--tracker_name', type=str,default='kys', help='Name of tracking method.') parser.add_argument('--tracker_param', type=str,default='default', help='Name of parameter file.')

parser.add_argument('--tracker_name', type=str,default='dimp', help='Name of tracking method.') parser.add_argument('--tracker_param', type=str,default='prdimp50', help='Name of parameter file.')