前面有人看了看了我的博客商汤PySot的配置使用(1)—siam跟踪算法demo、test、eval,问我怎么用pysot绘制ope曲线,大家可以参考下面的内容。
pysot的官方代码链接https://github.com/STVIR/pysot
一、步骤一:改eval.py的内容

主要是参考https://github.com/StrangerZhang/pysot-toolkit,将里面绘制曲线的代码加进去,大家可以对比一下STVIR/pysot的eval.py和StrangerZhang/pysot-toolkit的eval.py,就能清楚地看到我加了哪些内容。
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import os
import argparse
from glob import glob
from tqdm import tqdm
from multiprocessing import Pool
from toolkit.datasets import OTBDataset, UAVDataset, LaSOTDataset, \
VOTDataset, NFSDataset, VOTLTDataset
from toolkit.evaluation import OPEBenchmark, AccuracyRobustnessBenchmark, \
EAOBenchmark, F1Benchmark
from toolkit.visualization import draw_success_precision, draw_eao, draw_f1
parser = argparse.ArgumentParser(description='tracking evaluation')
parser.add_argument('--tracker_path', '-p', type=str,
help='tracker result path')
parser.add_argument('--dataset', '-d', type=str,
help='dataset name')
parser.add_argument('--num', '-n', default=1, type=int,
help='number of thread to eval')
parser.add_argument('--tracker_prefix', '-t', default='',
type=str, help='tracker name')
parser.add_argument('--show_video_level', '-s', dest='show_video_level',
action='store_true')
parser.add_argument('--vis', dest='vis', action='store_true')
parser.set_defaults(show_video_level=False)
args = parser.parse_args()
def main():
tracker_dir = os.path.join(args.tracker_path, args.dataset)
trackers = glob(os.path.join(args.tracker_path,
args.dataset,
args.tracker_prefix+'*'))
trackers = [os.path.basename(x) for x in trackers]
assert len(trackers) > 0
args.num =

本文详细介绍如何使用PYSOT库进行目标跟踪算法的评估,包括修改代码、配置环境、安装依赖及运行评估脚本的具体步骤,适用于OTB等常见数据集。
最低0.47元/天 解锁文章
281

被折叠的 条评论
为什么被折叠?



