单目标追踪 数据集学习笔记

1.数据集介绍

VTB数据集
里面有50个带有标注的数据集(https://img-blog.csdnimg.cn/ef0c188f73a14fd0a342845d71166f8b.png)

在这里插入图片描述
下载里面的任一视频
包含图片集和标注好的groundtruth_rect.txt
文件里面四列分别代表矩形框的 left_top(x,y) 和宽高（w,h）

2.数据集检查代码
1、提取box

#2023 1 February @JeffreySimon
import cv2 as cv
def get_rect(file_name):
    f=open(file_name,'r')
    lines=f.readlines()
    rect_boxes=[]
    for line in lines:
        print(line)
        line=line.replace("\n", "")
        rect_box=line.split(",")
        rect_box_int=[]
        for rb in rect_box:
            rect_box_int.append(int(rb))
        rect_boxes.append(rect_box_int)
        # print(rect_box)
    return rect_boxes

画框

#2023 1 February @JeffreySimon
import cv2 as cv
from getRect import get_rect as gr
file_name="data/Bird1/Bird1/groundtruth_rect.txt"  # 2-1 修改ground truth 的路径
path_root="data/Bird1/Bird1/img/" # 2-1 修改ground truth 的路径
pic_num=142

rect_boxes= gr(file_name)
def main():
    for i in range(1,pic_num):
        si=str(i)
        si=si.zfill(4)
        path=path_root+si+".jpg"
        print(path)
        img= cv.imread(path)
        r=rect_boxes[i]
        cv.rectangle(img,(r[0],r[1]),(r[0]+r[2],r[1]+r[3]),(255,0,0),2,lineType=cv.LINE_AA)
        cv.imshow("a",img)
        key=cv.waitKey(200)
        if key==27:
            exit(0)
if __name__ == '__main__':
     main()

代码和数据在