dataset/2detection.py

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)