Spaces:
Sleeping
Sleeping
File size: 491 Bytes
fa84113 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
""" Parser factory
Copyright 2020 Ross Wightman
"""
from .parser_coco import CocoParser
from .parser_voc import VocParser
from .parser_open_images import OpenImagesParser
def create_parser(name, **kwargs):
if name == 'coco':
parser = CocoParser(**kwargs)
elif name == 'voc':
parser = VocParser(**kwargs)
elif name == 'openimages':
parser = OpenImagesParser(**kwargs)
else:
assert False, f'Unknown dataset parser ({name})'
return parser
|