将Temple Color数据集标注转换为tracker benchmark所需格式

soldatJiang

于 2020-01-16 19:34:34 发布

阅读量1.2k

点赞数 5

分类专栏：视觉跟踪计算机视觉 python

本文链接：https://blog.youkuaiyun.com/u013806196/article/details/104010019

版权

视觉跟踪同时被 3 个专栏收录

7 篇文章

订阅专栏

计算机视觉

7 篇文章

订阅专栏

python

2 篇文章

订阅专栏

本文介绍如何将TempleColor128数据集的标注格式转换为OTB工具箱所需的格式，以便在写论文进行对比试验时使用。通过编写Python脚本，实现属性和跟踪框标注的轻松转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写论文对比试验需要在Temple Color 128数据集(http://www.dabi.temple.edu/~hbling/data/TColor-128/TColor-128.html)上跑，但其标注与OTB测评工具箱有所差别。
OTB工具箱所需的属性和跟踪框标注格式是这样的
在这里插入图片描述
而Temple Color 128提供的标注格式是这样的

编写脚本，轻松转换

import os
import shutil

CHALLENGE_ATTRIBUTE = ['IV','OPR','SV','OCC','DEF','MB','FM','IPR','OV','BC','LR']

if not os.path.exists('anno'):
    os.makedirs('anno')
    os.makedirs(os.path.join('anno','att'))

def process_att(att_filename):
    ret = []
    with open(att_filename) as f:
        for l in f.readlines():
            ret.append(l.strip())
    return ret
        
videos = os.listdir('Temple-color-128')

for v in videos:
    att_file = os.path.join('Temple-color-128',v,v+'_att.txt')
    gt_file = os.path.join('Temple-color-128',v,v+'_gt.txt')
    att_file_write = os.path.join('anno','att',v+'.txt')
    att_str = ''
    att = process_att(att_file)
    for c in CHALLENGE_ATTRIBUTE:
        if c in att:
            att_str += '1'
        else:
            att_str += '0'
        
        if c != 'LR':
            att_str += ','
    with open(att_file_write, 'w') as f:
        f.write(att_str)

    shutil.copy(gt_file, os.path.join('anno', v+'.txt'))