File size: 545 Bytes
3f3265f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | # obtain interactable dataset from semantics dataset
import json
import os
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--input', type=str, required=True)
parser.add_argument('--output', type=str, required=True)
args = parser.parse_args()
with open(args.input, 'r') as fr:
data = json.load(fr)
data['categories'] = [{'id': 1, 'name': 'interactable'}]
for annotation in data['annotations']:
annotation['category_id'] = 1
with open(args.output, 'w') as fw:
json.dump(data, fw, indent=4) |