Browse Source

changed default part surrogate type

Dimitri Korsch 5 years ago
parent
commit
f51fef1596
2 changed files with 2 additions and 1 deletions
  1. 1 0
      cvdatasets/annotations/__init__.py
  2. 1 1
      cvdatasets/dataset/image.py

+ 1 - 0
cvdatasets/annotations/__init__.py

@@ -41,5 +41,6 @@ class AnnotationType(BaseChoiceType):
 	INAT19_TEST = partial(INAT19_Annotations)
 
 	INAT20 = INAT20_Annotations
+	INAT20_TEST = partial(INAT20_Annotations)
 
 	Default = CUB200

+ 1 - 1
cvdatasets/dataset/image.py

@@ -38,7 +38,7 @@ class ImageWrapper(object):
 		parts=None,
 		mode="RGB",
 		part_rescale_size=None,
-		part_surrogate_type=SurrogateType.IMAGE,
+		part_surrogate_type=SurrogateType.MIDDLE,
 		center_cropped=True):
 
 		self.mode = mode