some small fixes. Updated the CUB200-GIS dataset definitions

cvdatasets/annotations/__init__.py

@@ -18,6 +18,7 @@ class AnnotationType(BaseChoiceType):
 	CUB200 = CUB_Annotations
 	CUB200_2FOLD = partial(CUB_Annotations)
 	CUB200_GOOGLE = partial(CUB_Annotations)
+	CUB200_GOOGLE_SEM = partial(CUB_Annotations)
 	NAB = NAB_Annotations
 	CARS = CARS_Annotations
 	DOGS = DOGS_Annotations

cvdatasets/annotations/base/parts_mixin.py

@@ -32,6 +32,8 @@ class PartsMixin(abc.ABC):
 		if self.part_type is not None:
 			new_kwargs["part_rescale_size"] = self.dataset_info.rescale_size
 
+		new_kwargs.update(kwargs)
+
 		return super(PartsMixin, self).check_dataset_kwargs(subset, **new_kwargs)
 
 	@property

scripts/info_files/info.yml

@@ -65,6 +65,11 @@ DATASETS:
     annotations: "ORIGINAL"
     n_classes: 200
 
+  CUB200_GOOGLE_SEM: &cub200_google_sem
+    folder: birds/cub200_google_images
+    annotations: "sem_selected"
+    n_classes: 200
+
 
   NAB:         &nabirds
     folder: birds/nabirds
@@ -176,6 +181,10 @@ PARTS:
     <<: *cub200_google
     <<: *parts_global
 
+  CUB200_GOOGLE_SEM_GLOBAL:
+    <<: *cub200_google_sem
+    <<: *parts_global
+
 
   CARS_GLOBAL:
     <<: *cars