From de3a91a7a0f1098b515c0bbe24b178467a6f2d70 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 16 Aug 2024 15:13:56 -0700 Subject: [PATCH] Add min_input_size of 128 for hieradet/sam2 --- timm/models/hieradet_sam2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/hieradet_sam2.py b/timm/models/hieradet_sam2.py index 464e331357..04cb1484a8 100644 --- 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