python download xf_Python 脚本下载 VOT 数据集

缺乏、安全感 2023-01-09 14:20 139阅读 0赞

Content:Approach to download vot-2019 dataset;

Approach to download vot-2019-rgbt dataset;

for the rgb-depth videos, or long-term videos, you can modify the code yourself to download them all.

Baiduyun links for vot-2018, vot-2019, vot-2018lt, vot-2019lt, vot-rgb-thermal dataset.

Approach to download vot-2019 dataset;

3. execute the following python script.

import json

import os

import wget,tarfile,zipfile

vot\_2019\_path = '/home/wangxiao/Documents/dataset/vot2019/' \# object file

json\_path = '/home/wangxiao/Downloads/description.json' \# vot 2019 json file

anno\_vot = 'vot2019' \# vot2019 or vot2018 or vot2017

with open(json\_path,'r') as fd:

vot\_2019 = json.load(fd)

home\_page = vot\_2019\['homepage'\]

for i,sequence in enumerate(vot\_2019\['sequences'\]):

print('download the \{\} sequences'.format(i+1))

annotations = sequence\['annotations'\]\['url'\]

data\_url = sequence\['channels'\]\['color'\]\['url'\].split('../../')\[-1\]

name = annotations.split('.')\[0\]

file\_name = annotations.split('.')\[0\] + '.zip'

down\_annotations = os.path.join(home\_page,anno\_vot,'main',annotations)

down\_data\_url = os.path.join(home\_page,data\_url)

image\_output\_name = os.path.join(vot\_2019\_path,name,'color',file\_name)

anno\_output\_name = os.path.join(vot\_2019\_path,name,file\_name)

out\_dir = os.path.dirname(anno\_output\_name)

if os.path.exists(out\_dir) == False:

os.mkdir(out\_dir)

\# annotations download and unzip and remove it

wget.download(down\_annotations, anno\_output\_name)

print('loading \{\} annotation'.format(name))

\# unzip

file\_zip = zipfile.ZipFile(anno\_output\_name,'r')

for file in file\_zip.namelist():

file\_zip.extract(file, out\_dir)

print('extract annotation \{\}/\{\}'.format(name,file))

file\_zip.close()

os.remove(anno\_output\_name)

print('remove annotation \{\}.zip'.format(name))

\# image download and unzip ad remove it

out\_dir = os.path.dirname(image\_output\_name)

if os.path.exists(out\_dir) == False:

os.mkdir(out\_dir)

wget.download(down\_data\_url,image\_output\_name)

print('loading \{\} sequence'.format(name))

file\_zip = zipfile.ZipFile(image\_output\_name,'r')

for file in file\_zip.namelist():

file\_zip.extract(file,out\_dir)

print('extract image \{\}'.format(file))

file\_zip.close()

os.remove(image\_output\_name)

print('remove image file \{\}.zip'.format(name))

print('sequence \{\} Completed!'.format(i+1))Approach to download vot-2019-rgb-thermal dataset;![Image 1][]

3. python download\_vot2019\_rgbt\_dataet\_script.py

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#　　　　　　　　　　 　Download the VOT2019-rgbt-thermal dataset\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#import json

import os

import wget,tarfile,zipfile

import pdb

vot\_2019\_path = '/home/wangxiao/Documents/dataset/vot2019\_rgbt\_dataset/' \# object file

json\_path = '/home/wangxiao/Downloads/description\_rgbt.json' \# vot 2019 json file

anno\_vot = 'vot2019' \# vot2019 or vot2018 or vot2017

with open(json\_path,'r') as fd:

vot\_2019 = json.load(fd)

home\_page = vot\_2019\['homepage'\]

for i,sequence in enumerate(vot\_2019\['sequences'\]):

print('download the\{\}sequences'.format(i+1))

annotations = sequence\['annotations'\]\['url'\]

rgb\_data\_url = sequence\['channels'\]\['color'\]\['url'\].split('../')\[-1\]

ir\_data\_url = sequence\['channels'\]\['ir'\]\['url'\].split('../')\[-1\]

name = annotations.split('.')\[0\]

file\_name = annotations.split('.')\[0\] + '.zip'

down\_annotations = os.path.join(home\_page, anno\_vot,'rgbtir/meta', annotations)

down\_rgb\_data\_url = os.path.join(home\_page, anno\_vot, 'rgbtir', rgb\_data\_url)

down\_ir\_data\_url = os.path.join(home\_page, anno\_vot, 'rgbtir', ir\_data\_url)

rgb\_image\_output\_name = os.path.join(vot\_2019\_path, name, 'color', file\_name)

ir\_image\_output\_name = os.path.join(vot\_2019\_path, name, 'ir', file\_name)

anno\_output\_name = os.path.join(vot\_2019\_path, name, file\_name)

out\_dir = os.path.dirname(anno\_output\_name)

if os.path.exists(out\_dir) == False:

os.mkdir(out\_dir)

\# pdb.set\_trace()

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\#\#\#\# Download Annotation files

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\# annotations download and unzip and remove it

wget.download(down\_annotations, anno\_output\_name)

print('loading\{\}annotation'.format(name))

\# unzip

file\_zip = zipfile.ZipFile(anno\_output\_name,'r')

for file in file\_zip.namelist():

file\_zip.extract(file, out\_dir)

print('extract annotation\{\}/\{\}'.format(name,file))

file\_zip.close()

os.remove(anno\_output\_name)

print('remove annotation\{\}.zip'.format(name))

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\#\#\#\# Download Color images

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\# image download and unzip ad remove it

out\_dir = os.path.dirname(rgb\_image\_output\_name)

if os.path.exists(out\_dir) == False:

os.mkdir(out\_dir)

\# pdb.set\_trace()

wget.download(down\_rgb\_data\_url, rgb\_image\_output\_name)

print('loading\{\}sequence --- rgb files ...'.format(name))

file\_zip = zipfile.ZipFile(rgb\_image\_output\_name,'r')

for file in file\_zip.namelist():

file\_zip.extract(file,out\_dir)

print('extract image\{\}'.format(file))

file\_zip.close()

os.remove(rgb\_image\_output\_name)

print('remove image file\{\}.zip'.format(name))

print('sequence\{\}Completed!'.format(i+1))

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\#\#\#\# Download Infrared images

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#

\# image download and unzip ad remove it

out\_dir = os.path.dirname(ir\_image\_output\_name)

if os.path.exists(out\_dir) == False:

os.mkdir(out\_dir)

\# pdb.set\_trace()

wget.download(down\_ir\_data\_url, ir\_image\_output\_name)

print('loading\{\}sequence --- infrared files ...'.format(name))

file\_zip = zipfile.ZipFile(ir\_image\_output\_name, 'r')

for file in file\_zip.namelist():

file\_zip.extract(file,out\_dir)

print('extract image\{\}'.format(file))

file\_zip.close()

os.remove(ir\_image\_output\_name)

print('remove image file\{\}.zip'.format(name))

print('sequence\{\}Completed!'.format(i+1))

print("=====================================================================")

print(" Great!!! All videos have been dowloaded ......")

print("=====================================================================")

4. Then, you can download videos automatically.![Image 1][]![Image 1][]

If you just want the dataset, well, I have downloaded all the videos and uploaded them into the Baiduyun drive. You can take it away, and No Thanks!Reference:

[Image 1]: