Dimitri Korsch 6 жил өмнө
parent
commit
bb2742bf13

+ 2 - 1
nabirds/__init__.py

@@ -1,4 +1,5 @@
 from .dataset import Dataset, ImageWrapperDataset
-from .annotations import NAB_Annotations, CUB_Annotations
+from .annotations import *
+from .utils import _MetaInfo
 
 __version__ = "0.3.4"

+ 2 - 0
nabirds/annotations/__init__.py

@@ -4,6 +4,8 @@ from .cars import CARS_Annotations
 from .inat import INAT19_Annotations
 from .flowers import FLOWERS_Annotations
 
+from .base import BaseAnnotations
+
 from cvargparse.utils import BaseChoiceType
 from functools import partial