Spaces:
Sleeping
Sleeping
File size: 979 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 |
# --------------------------------------------------------
# SiamMask
# Licensed under The MIT License
# Written by Qiang Wang (wangqiang2015 at ia.ac.cn)
# --------------------------------------------------------
import json
print('load json (raw ytb_vos info), please wait 10 seconds~')
ytb_vos = json.load(open('instances_train.json', 'r'))
snippets = dict()
for k, v in ytb_vos.items():
video = dict()
for i, o in enumerate(list(v)):
obj = v[o]
snippet = dict()
trackid = "{:02d}".format(i)
for frame in obj:
file_name = frame['file_name']
frame_name = '{:06d}'.format(int(file_name.split('/')[-1]))
bbox = frame['bbox']
bbox[2] += bbox[0]
bbox[3] += bbox[1]
snippet[frame_name] = bbox
video[trackid] = snippet
snippets['train/'+k] = video
train = snippets
json.dump(train, open('train.json', 'w'), indent=4, sort_keys=True)
print('done!')
|