Detic / tools /get_coco_zeroshot_oriorder.py
AK391
files
159f437
# Copyright (c) Facebook, Inc. and its affiliates.
import argparse
import json
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--data_path', default='datasets/coco/annotations/instances_val2017_unseen_2.json')
parser.add_argument('--cat_path', default='datasets/coco/annotations/instances_val2017.json')
args = parser.parse_args()
print('Loading', args.cat_path)
cat = json.load(open(args.cat_path, 'r'))['categories']
print('Loading', args.data_path)
data = json.load(open(args.data_path, 'r'))
data['categories'] = cat
out_path = args.data_path[:-5] + '_oriorder.json'
print('Saving to', out_path)
json.dump(data, open(out_path, 'w'))