Browse Source

added noisy iNat2020 class

Dimitri Korsch 5 years ago
parent
commit
e5909799a5
2 changed files with 10 additions and 1 deletions
  1. 1 0
      cvdatasets/annotations/__init__.py
  2. 9 1
      scripts/info_files/info.yml

+ 1 - 0
cvdatasets/annotations/__init__.py

@@ -44,5 +44,6 @@ class AnnotationType(BaseChoiceType):
 	INAT20_TEST = partial(INAT20_Annotations)
 	INAT20_IN_CLASS = partial(INAT20_Annotations)
 	INAT20_OUT_CLASS = partial(INAT20_Annotations)
+	INAT20_NOISY_IN_CLASS = partial(INAT20_Annotations)
 
 	Default = CUB200

+ 9 - 1
scripts/info_files/info.yml

@@ -105,10 +105,14 @@ DATASETS:
     annotations: "2020/PLAIN"
     n_classes: 200
 
-  INAT20_IN_CLASS: &inat20_in_class
+  INAT20_IN_CLASS:         &inat20_in_class
     <<: *inat20
     annotations: "2020/IN_CLASS"
 
+  INAT20_NOISY_IN_CLASS: &inat20_noisy_in_class
+    <<: *inat20
+    annotations: "2020/NOISY_IN_CLASS"
+
   INAT20_OUT_CLASS:         &inat20_out_class
     <<: *inat20
     annotations: "2020/OUT_CLASS"
@@ -287,6 +291,10 @@ PARTS:
     <<: *inat20_out_class
     <<: *parts_global
 
+  INAT20_NOISY_IN_CLASS_GLOBAL:
+    <<: *inat20_noisy_in_class
+    <<: *parts_global
+
   HED_GLOBAL:
     <<: *hed
     <<: *parts_global