How to get input shape for each pretrained model?

[JohnG0024]

timm.list_models() lists all the pretrained models, but how do I get the input shape for each model?

[TorbenSDJohansen]

To get the default input shape, one way would be to run the following code:

import timm

model_to_input_size = {}

for model in timm.list_models():
    default_cfg = timm.models.registry._model_default_cfgs.get(model, None)
    input_size = default_cfg['input_size'] if default_cfg else None
    model_to_input_size[model] = input_size

Note that very few models do not have a default_cfg. These are, e.g., deprecated models.

[TorbenSDJohansen]

One small note. If you are only interested in pretrained models, be sure to change timm.list_models() to timm.list_models(pretrained=True).

[JohnG0024]

Thank you so much.