__init__.py 417 B

12345678910111213141516
  1. from .cub import CUB_Annotations
  2. from .nab import NAB_Annotations
  3. from .cars import CARS_Annotations
  4. from .inat import INAT19_Annotations
  5. from cvargparse.utils import BaseChoiceType
  6. from functools import partial
  7. class AnnotationType(BaseChoiceType):
  8. CUB200 = CUB_Annotations
  9. NAB = NAB_Annotations
  10. CARS = CARS_Annotations
  11. INAT19 = INAT19_Annotations
  12. INAT19_MINI = partial(INAT19_Annotations)
  13. Default = CUB200