@@ -41,5 +41,6 @@ class AnnotationType(BaseChoiceType):
INAT19_TEST = partial(INAT19_Annotations)
INAT20 = INAT20_Annotations
+ INAT20_TEST = partial(INAT20_Annotations)
Default = CUB200
@@ -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