Add min_input_size of 128 for hieradet/sam2

huggingface · Aug 16, 2024 · de3a91a · de3a91a
1 parent 0b05122
commit de3a91a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/hieradet_sam2.py b/timm/models/hieradet_sam2.py
@@ -517,7 +517,7 @@ def _cfg(url='', **kwargs):
     return {
         'url': url,
         'num_classes': 0, 'input_size': (3, 896, 896), 'pool_size': (28, 28),
-        'crop_pct': 1.0, 'interpolation': 'bicubic',
+        'crop_pct': 1.0, 'interpolation': 'bicubic', 'min_input_size': (3, 128, 128),
         'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
         'first_conv': 'patch_embed.proj', 'classifier': 'head.fc',
         **kwargs