ubuntu16.04下如何训练PySOT(五):补充,安装 pysot-toolkit ,测试评估模型

it2026-02-03  1

安装pysot-toolkit

在pysot-master文件目录打开终端,输入以下命令

git clone https://github.com/StrangerZhang/pysot-toolkit cd pysot-toolkit pip install -r requirements.txt cd pysot/utils/ python setup.py build_ext --inplace

在安装requirements的工具包时,安装glob会出现错误,将glob改为glob2重新安装即可

将pysot-toolkit改名为toolkit

测试

修改文件tools/test.py

修改文件中的20行

修改前 from toolkit.datasets import DatasetFactory 修改后 from toolkit.pysot.datasets import DatasetFactory

修改文件中的21行

修改前 from toolkit.utils.region import vot_overlap, vot_float2str 修改后 from toolkit.pysot.utils.region import vot_overlap, vot_float2str

测试集是采用OTB100,在tools目录下打开终端,输入以下命令,开始测试

python -u test.py --snapshot model.pth --dataset OTB100 --config ../experiments/siamrpn_r50_l234_dwxcorr_otb/config.yaml

测试的结果保存在以下目录tools/results/数据集名称/model

如下

评估

修改文件tools/eval.py

修改文件第12行

修改前 from toolkit.datasets import OTBDataset, UAVDataset, LaSOTDataset, \ VOTDataset, NFSDataset, VOTLTDataset 修改后 from toolkit.pysot.datasets import OTBDataset, UAVDataset, LaSOTDataset, \ VOTDataset, NFSDataset, VOTLTDataset

修改文件第14行

修改前 from toolkit.evaluation import OPEBenchmark, AccuracyRobustnessBenchmark, \ EAOBenchmark, F1Benchmark 修改后 from toolkit.pysot.evaluation import OPEBenchmark, AccuracyRobustnessBenchmark, \ EAOBenchmark, F1Benchmark

在tools命令行下输入以下命令,开始评估,评估数据集采用OTB100数据集

python eval.py --tracker_path ./results --dataset OTB100 --num 1 --tracker_prefix 'model'

 

最新回复(0)