added 2Fold-CUB200 Dataset

cvdatasets/annotations/__init__.py

@@ -13,6 +13,7 @@ from functools import partial
 
 class AnnotationType(BaseChoiceType):
 	CUB200 = CUB_Annotations
+	CUB200_2FOLD = partial(CUB_Annotations)
 	NAB = NAB_Annotations
 	CARS = CARS_Annotations
 	DOGS = DOGS_Annotations

scripts/info_files/info.yml

@@ -55,6 +55,11 @@ DATASETS:
     annotations: "ORIGINAL"
     n_classes: 200
 
+  CUB200_2FOLD:   &cub200_2fold
+    folder: birds/cub200_2fold
+    annotations: "ORIGINAL"
+    n_classes: 200
+
   NAB:         &nabirds
     folder: birds/nabirds
     annotations: "ORIGINAL"
@@ -143,6 +148,10 @@ PART_TYPES:
 PARTS:
 
   #### No Parts Annotations
+  CUB200_2FOLD_GLOBAL:
+    <<: *cub200_2fold
+    <<: *parts_global
+
   CUB200_GLOBAL:
     <<: *cub200
     <<: *parts_global
@@ -184,6 +193,22 @@ PARTS:
   #### With Parts Annotations
 
 
+  ####################################
+  # CUB200-2011 2-Fold training set
+  ####################################
+
+  CUB200_2FOLD_L1_pred:
+    <<: *cub200_2fold
+    <<: *parts_l1p
+    annotations: L1_pred
+    rescale_size: !!int 427
+
+  CUB200_2FOLD_L1_full:
+    <<: *cub200_2fold
+    <<: *parts_l1f
+    annotations: L1_full
+    rescale_size: !!int 427
+
   ####################################
   # CUB200-2011
   ####################################
@@ -209,14 +234,14 @@ PARTS:
   CUB200_L1_pred:
     <<: *cub200
     <<: *parts_l1p
-    annotations: cub200_11_L1_pred
-    rescale_size: !!int 427
+    annotations: cub200_11_L1_pred_old
+    # rescale_size: !!int 427
 
   CUB200_L1_full:
     <<: *cub200
     <<: *parts_l1f
-    annotations: cub200_11_L1_full
-    rescale_size: !!int 427
+    annotations: cub200_11_L1_full_old
+    # rescale_size: !!int 427
 
   CUB200_L1_pred_15:
     <<: *cub200