使用pysot-toolkit绘制跟踪算法雷达图

Brilliant Cheng

已于 2022-06-12 20:28:46 修改

阅读量2.2k

点赞数 8

分类专栏：目标跟踪文章标签：目标跟踪人工智能计算机视觉

于 2022-06-12 20:28:06 首次发布

本文链接：https://blog.csdn.net/qq_30287833/article/details/125249403

版权

目标跟踪专栏收录该内容

3 篇文章 0 订阅

订阅专栏

使用pysot-toolkit绘制跟踪算法雷达图

说明：使用pysot-toolkit生成评估目标跟踪算法的雷达图

pysot-toolkit： StrangerZhang/pysot-toolkit: Python Single Object Tracking Evaluation (github.com)

为了更好的对比不同的目标跟踪算法在应对不同挑战时的表现，使用雷达图的形式可以更好的对比不同算法之间的优势和差异，先看看效果图

在这里插入图片描述

生成雷达图是为了看跟踪算法在不同挑战下的得分情况，因此需要在具有不同挑战的数据集下对跟踪器进行测试，然后用该工具进行对比不同的跟踪算法。

具有不同挑战的数据集主要有：VOT，OTB， LaSOT

安装：参考README.md

cd path/pysot-toolkit
pip install -r requirements.txt
cd pysot/utils
python setup.py build_ext --inplace

将以下代码做些许的改动：

./pysot/evaluation/eao_benchmark.py

在这里插入图片描述

./bin/eval.py

在这里插入图片描述

依赖项：LaTex

readme中说了，要是画图的话需要安装LaTex，在这里我没来得及安装，所以运行出错，提示未找到LaTeX

RuntimeError: Failed to process string with tex because latex could not be found

其实简单的没有LaTeX也是可以的，修改./pysot/visualization/draw_eao.py：

在这里插入图片描述

可以生成对应的图，但就是下面图例的显示会存在问题。图中我使用了6个跟踪算法，但是只显示了部分，所以如果有问题的话，建议还是下载LaTex吧。

这个问题的解决可以参考：@张小波：学习 PySOT(2)（PySOT-toolkit、对比、画图）

运行 ./bin/eval.py 时的参数配置

--dataset_dir
/home/user2/Documents/dataset/VOT2018  # 数据集地址
--dataset
VOT2018  # 数据集名称
--tracker_result_dir
/home/user2/Documents/code/ChengH/pysot-toolkit/results/VOT2018  # 测试结果存放的路径
--trackers   # 跟踪器的名称
Ocean
SiamBAN
SiamFCpp
SiamRCNN
SiamRPNpp
TrDiMP

这里主要说下存放结果的路径

# 文件路径./results
results
-----VOT2018
			-----Ocean
						-----baseline
			-----SiamBAN
			-----SiamFCpp
			-----SiamRCNN
			-----SiamRPNpp
			-----TrDiMP