Explorar o código

added noisy out-class iNat20 dataset

Dimitri Korsch %!s(int64=5) %!d(string=hai) anos
pai
achega
c89048d04f
Modificáronse 2 ficheiros con 19 adicións e 0 borrados
  1. 1 0
      cvdatasets/annotations/__init__.py
  2. 18 0
      scripts/info_files/info.yml

+ 1 - 0
cvdatasets/annotations/__init__.py

@@ -46,5 +46,6 @@ class AnnotationType(BaseChoiceType):
 	INAT20_OUT_CLASS = partial(INAT20_Annotations)
 	INAT20_NOISY_IN_CLASS = partial(INAT20_Annotations)
 	INAT20_U_OUT_CLASS = partial(INAT20_Annotations)
+	INAT20_NOISY_OUT_CLASS = partial(INAT20_Annotations)
 
 	Default = CUB200

+ 18 - 0
scripts/info_files/info.yml

@@ -49,6 +49,7 @@ MODELS:
 
 ############ Existing Datasets
 DATASETS:
+
   IMAGENET:         &inet
     folder: ImageNet
     annotations: "BJOERN"
@@ -85,6 +86,7 @@ DATASETS:
     annotations: "RESIZED"
     n_classes: 500
 
+
   CARS:         &cars
     folder: cars
     annotations: "ORIGINAL"
@@ -117,6 +119,14 @@ DATASETS:
     <<: *inat20
     annotations: "2020/OUT_CLASS"
 
+  INAT20_NOISY_OUT_CLASS:         &inat20_noisy_out_class
+    <<: *inat20
+    annotations: "2020/NOISY_OUT_CLASS"
+
+  INAT20_U_OUT_CLASS:         &inat20_u_out_class
+    <<: *inat20
+    annotations: "2020/U_OUT_CLASS"
+
   INAT20_TEST:    &inat20_test
     <<: *inat20
     annotations: "2020/TEST"
@@ -291,6 +301,14 @@ PARTS:
     <<: *inat20_out_class
     <<: *parts_global
 
+  INAT20_U_OUT_CLASS_GLOBAL:
+    <<: *inat20_u_out_class
+    <<: *parts_global
+
+  INAT20_NOISY_OUT_CLASS_GLOBAL:
+    <<: *inat20_noisy_out_class
+    <<: *parts_global
+
   INAT20_NOISY_IN_CLASS_GLOBAL:
     <<: *inat20_noisy_in_class
     <<: *parts_global