目标跟踪常见训练数据集格式

coco

-annotations

• captions_train2017.json
• captions_val2017.json
• instances_train2017.json
• instances_val2017.json
• person_keypoints_train2017.json
• person_keypoints_val2017.json

-train2017

• 0000000001.jpg
• ...
• 0000581929.jpg

got10k

-train

        -GOT-10_Train_000001

•         000000001.jpg
•         ...
•         000000120.jpg
•         absence.lable
•         cover.label
•         cut_by_image.lablel
•         groundtruth.txt
•         meta_info.ini

        ...

        -GOT-10_Train_009335

• list.txt

-test

        -GOT-10_Test_000001

        ...

        -GOT-10_Test_000180

• list.txt

-val

        -GOT-10_val_000001

        ...

        -GOT-10_val_000180

• list.txt

# GOT-10k
import os
import zipfile
import tqdm

got10k_path = '/home/XXX/Desktop/data_ssd/got10k/train_data'
files = os.listdir(got10k_path)
for file in tqdm.tqdm(files):
    if file[-3:] == 'zip':
        zip_file = os.path.join(got10k_path, file)
        fz = zipfile.ZipFile(zip_file, 'r')
        for i in fz.namelist():
            fz.extract(i, os.path.join(got10k_path, file[:-4]))
        os.remove(zip_file)

lasot

-airplane

        -airplane-1

                -img

•                         000000001.jpg
•                         ...
•                         000002788.jpg
•                 full_occlusion.txt
•                 groundtruth
•                 nlp.txt
•                 out-of-view.txt

        ...

        -airplane-20

...

-zebra

trackingnet

-TRAIN_0

        -anno

•                 3TIfnTSM6c_2.txt
•                 ...

        -frames

                -3TIfnTSM6c_2

•                         0.jpg
•                         ...
•                         469.jpg

                ...

...

-TRAIN_11

-TEST

# TrackingNet
import os
import zipfile
import tqdm

trackingnet_path = '/home/XXX/Desktop/data_ssd/trackingnet'
files = os.listdir(trackingnet_path)
for file in tqdm.tqdm(files):
    print('Ready to unzip files in '+file+'...')
    zips_path = os.listdir(os.path.join(trackingnet_path, file, 'zips'))
    for zip in tqdm.tqdm(zips_path):
        if os.path.join(trackingnet_path, file, 'zips', zip)[-3:] == 'zip':
            fz = zipfile.ZipFile(os.path.join(trackingnet_path, file, 'zips', zip), 'r')
            for i in fz.namelist():
                fz.extract(i, os.path.join(trackingnet_path, file, 'zips', zip)[:-4])
            os.remove(os.path.join(trackingnet_path, file, 'zips', zip))