taskswithcode/DeticChatGPT

1# Copyright (c) Facebook, Inc. and its affiliates.2import argparse3import json4 5if __name__ == '__main__':6    parser = argparse.ArgumentParser()7    parser.add_argument('--data_path', default='datasets/coco/annotations/instances_val2017_unseen_2.json')8    parser.add_argument('--cat_path', default='datasets/coco/annotations/instances_val2017.json')9    args = parser.parse_args()10    print('Loading', args.cat_path)11    cat = json.load(open(args.cat_path, 'r'))['categories']12 13    print('Loading', args.data_path)14    data = json.load(open(args.data_path, 'r'))15    data['categories'] = cat16    out_path = args.data_path[:-5] + '_oriorder.json'17    print('Saving to', out_path)18    json.dump(data, open(out_path, 'w'))19