Spaces:
Paused
Paused
File size: 758 Bytes
6e14436 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# Copyright (c) Facebook, Inc. and its affiliates.
import argparse
import json
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--ann', default='datasets/lvis/lvis_v1_train.json')
args = parser.parse_args()
print('Loading', args.ann)
data = json.load(open(args.ann, 'r'))
catid2freq = {x['id']: x['frequency'] for x in data['categories']}
print('ori #anns', len(data['annotations']))
exclude = ['r']
data['annotations'] = [x for x in data['annotations'] \
if catid2freq[x['category_id']] not in exclude]
print('filtered #anns', len(data['annotations']))
out_path = args.ann[:-5] + '_norare.json'
print('Saving to', out_path)
json.dump(data, open(out_path, 'w'))
|