| # 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) |