Преглед на файлове

added new dataset type... this really need a rework

Dimitri Korsch преди 5 години
родител
ревизия
6a19578df0
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      cvdatasets/annotations/__init__.py

+ 2 - 0
cvdatasets/annotations/__init__.py

@@ -48,4 +48,6 @@ class AnnotationType(BaseChoiceType):
 	INAT20_U_OUT_CLASS = partial(INAT20_Annotations)
 	INAT20_NOISY_OUT_CLASS = partial(INAT20_Annotations)
 
+	IMAGENET_TOP_INAT20 = partial(INET_Annotations)
+
 	Default = CUB200