Spaces:
No application file
No application file
File size: 1,419 Bytes
907462b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | import sys
import argparse
import multiprocessing as mp
from pathlib import Path
sys.path.insert(0, '.')
from isegm.utils.exp import load_config_file
from scripts.annotations_conversion import openimages, ade20k, coco_lvis
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('dataset', choices=['openimages', 'ade20k', 'coco_lvis'], help='')
parser.add_argument('--split', nargs='+', choices=['train', 'val', 'test'], type=str, default=['train', 'val'],
help='')
parser.add_argument('--n-jobs', type=int, default=10)
parser.add_argument('--config-path', type=str, default='./config.yml',
help='The path to the config file.')
args = parser.parse_args()
cfg = load_config_file(args.config_path, return_edict=True)
return args, cfg
def main():
mp.set_start_method('spawn')
args, cfg = parse_args()
for split in args.split:
if args.dataset == 'openimages':
openimages.create_annotations(Path(cfg.OPENIMAGES_PATH), dataset_split=split)
elif args.dataset == 'ade20k' and split != 'test':
ade20k.create_annotations(Path(cfg.ADE20K_PATH), dataset_split=split, n_jobs=args.n_jobs)
elif args.dataset == 'coco_lvis':
coco_lvis.create_annotations(Path(cfg.LVIS_PATH), Path(cfg.COCO_PATH), dataset_split=split)
if __name__ == '__main__':
main()
|