mlx_mimm_multiscale_feature.py

# uvx --with "git+https://github.com/travishsu/mimm.git" --with numpy ipython
import mlx.core as mx
from mimm import get_model, list_models

model = get_model('resnet50', pretrained=True)
img = mx.random.normal((3, 256, 256, 3))
features = model.features(img)
for layer, feature in features.items():
    print(layer, feature.shape)

# layer1 (3, 64, 64, 256)
# layer2 (3, 32, 32, 512)
# layer3 (3, 16, 16, 1024)
# layer4 (3, 8, 8, 2048)