Ver código fonte

added test subset for iNaturalist 2019

Dimitri Korsch 6 anos atrás
pai
commit
b19461d990

+ 1 - 0
nabirds/annotations/__init__.py

@@ -12,5 +12,6 @@ class AnnotationType(BaseChoiceType):
 	CARS = CARS_Annotations
 	INAT19 = INAT19_Annotations
 	INAT19_MINI = partial(INAT19_Annotations)
+	INAT19_TEST = partial(INAT19_Annotations)
 
 	Default = CUB200

+ 1 - 1
nabirds/annotations/inat.py

@@ -14,7 +14,7 @@ class INAT19_Annotations(BaseAnnotations):
 	@property
 	def meta(self):
 		info = _MetaInfo(
-			images_folder="",
+			images_folder="images",
 			content="trainval2019.json",
 			val_content="val2019.json",
 			# train_content="train2019.json",

+ 22 - 1
scripts/info_files/info.yml

@@ -69,6 +69,10 @@ DATASETS:
     <<: *inat19
     annotations: "2019_small"
 
+  INAT19_TEST:    &inat19_test
+    <<: *inat19
+    annotations: "2019_test"
+
 ############ Existing Part Annotations and Part Features
 ### feature file name composition:
 # ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features
@@ -136,6 +140,11 @@ PARTS:
     <<: *parts_global
     feature_suffix: .mini
 
+  INAT19_TEST_GLOBAL:
+    <<: *inat19_test
+    <<: *parts_global
+    feature_suffix: .test
+
   #### No Parts Annotations
   CUB200_GLOBAL:
     <<: *cub200
@@ -190,6 +199,18 @@ PARTS:
   INAT19_MINI_L1_full:
     <<: *inat19_mini
     <<: *parts_l1f
-    feature_suffix: _5parts_L1_full.mini
     annotations: 2019_small_L1_full
+    feature_suffix: _5parts_L1_full.mini
+
+  INAT19_TEST_L1_pred:
+    <<: *inat19_test
+    <<: *parts_l1p
+    annotations: 2019_test_L1_pred
+    feature_suffix: _5parts_L1_pred.test
+
+  INAT19_TEST_L1_full:
+    <<: *inat19_test
+    <<: *parts_l1f
+    annotations: 2019_test_L1_full
+    feature_suffix: _5parts_L1_full.test