File size: 530 Bytes
3f3265f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import json
import argparse
import os
parser = argparse.ArgumentParser()
parser.add_argument('--input', type=str, required=True)
parser.add_argument('--output', type=str, required=True)
args = parser.parse_args()
if __name__ == '__main__':
with open(args.input, 'r') as f:
data = json.load(f)
for ann in data['annotations']:
ann.pop('segmentation')
ann['iscrowd'] = 0
os.makedirs(os.path.dirname(args.output), exist_ok=True)
with open(args.output, 'w') as f:
json.dump(data, f) |