浏览代码

added noisy iNat2020 class

Dimitri Korsch 5 年之前
父节点
当前提交
e5909799a5
共有 2 个文件被更改,包括 10 次插入1 次删除
  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