Spaces:
Sleeping
Sleeping
| # -------------------------------------------------------- | |
| # 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!') | |