尧图建网站 尧图建网站 YAOTU WEB BUILD 免费咨询
ARTICLE DETAIL

资讯详情

深耕网站建设与建站编程的一线实战洞察。

pytracking 部署笔记

pytracking 部署笔记 目录参数import sys sys.path.insert(0, rE:/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 rC:\Users\ChanJing-01\Videos\yumao\f_400 gt_path rC:\Users\ChanJing-01\Videos\yumao\f_400\gt.txt self.name my_own_dataset self.base_path rE:/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,, dtypenp.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(namemy_sequence, framesself.img_list, datasetself, # 传入自己作为dataset ground_truth_rectself.gt_rect, object_classunknown, target_visibleNone) 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, debug0)参数E:\project\track\pytracking-master\pytracking\parameter\dimp\dimp18.py本质是找代码路径parser.add_argument(--tracker_name, typestr,defaultkys, helpName of tracking method.) parser.add_argument(--tracker_param, typestr,defaultdefault, helpName of parameter file.)parser.add_argument(--tracker_name, typestr,defaultdimp, helpName of tracking method.) parser.add_argument(--tracker_param, typestr,defaultprdimp50, helpName of parameter file.)
返回列表