123456789101112131415161718192021222324252627282930313233 |
- import argparse
- import os.path as osp
- from mmengine.fileio import dump, load
- def parse_args():
- parser = argparse.ArgumentParser(
- description='Generate COCO test image information '
- 'for COCO panoptic segmentation.')
- parser.add_argument('data_root', help='Path to COCO annotation directory.')
- args = parser.parse_args()
- return args
- def main():
- args = parse_args()
- data_root = args.data_root
- val_info = load(osp.join(data_root, 'panoptic_val2017.json'))
- test_old_info = load(osp.join(data_root, 'image_info_test-dev2017.json'))
- # replace categories from image_info_test-dev2017.json
- # with categories from panoptic_val2017.json which
- # has attribute `isthing`.
- test_info = test_old_info
- test_info.update({'categories': val_info['categories']})
- dump(test_info, osp.join(data_root,
- 'panoptic_image_info_test-dev2017.json'))
- if __name__ == '__main__':
- main()
|