Spaces:
Running
on
T4
Running
on
T4
File size: 2,046 Bytes
d4b77ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
# --------------------------------------------------------
# SiamMask
# Licensed under The MIT License
# Written by Qiang Wang (wangqiang2015 at ia.ac.cn)
# --------------------------------------------------------
from os.path import join, isdir
from os import mkdir
import glob
import xml.etree.ElementTree as ET
import json
js = {}
VID_base_path = './ILSVRC2015'
ann_base_path = join(VID_base_path, 'Annotations/DET/train/')
sub_sets = ('ILSVRC2013_train', 'ILSVRC2013_train_extra0', 'ILSVRC2013_train_extra1', 'ILSVRC2013_train_extra2', 'ILSVRC2013_train_extra3', 'ILSVRC2013_train_extra4', 'ILSVRC2013_train_extra5', 'ILSVRC2013_train_extra6', 'ILSVRC2013_train_extra7', 'ILSVRC2013_train_extra8', 'ILSVRC2013_train_extra9', 'ILSVRC2013_train_extra10', 'ILSVRC2014_train_0000', 'ILSVRC2014_train_0001','ILSVRC2014_train_0002','ILSVRC2014_train_0003','ILSVRC2014_train_0004','ILSVRC2014_train_0005','ILSVRC2014_train_0006')
for sub_set in sub_sets:
sub_set_base_path = join(ann_base_path, sub_set)
if 'ILSVRC2013_train' == sub_set:
xmls = sorted(glob.glob(join(sub_set_base_path, '*', '*.xml')))
else:
xmls = sorted(glob.glob(join(sub_set_base_path, '*.xml')))
n_imgs = len(xmls)
for f, xml in enumerate(xmls):
print('subset: {} frame id: {:08d} / {:08d}'.format(sub_set, f, n_imgs))
xmltree = ET.parse(xml)
objects = xmltree.findall('object')
video = join(sub_set, xml.split('/')[-1].split('.')[0])
for id, object_iter in enumerate(objects):
bndbox = object_iter.find('bndbox')
bbox = [int(bndbox.find('xmin').text), int(bndbox.find('ymin').text),
int(bndbox.find('xmax').text), int(bndbox.find('ymax').text)]
frame = '%06d' % (0)
obj = '%02d' % (id)
if video not in js:
js[video] = {}
if obj not in js[video]:
js[video][obj] = {}
js[video][obj][frame] = bbox
json.dump(js, open('train.json', 'w'), indent=4, sort_keys=True)
|