mimm 0.0.2

MLX Image Models

MLX Image Models

Introduction

This repository contains implementations for image models in MLX with functions to load pretrained weights from PyTorch. Example training code is provided in mimm/scripts.

Installation

pip install mimm

Usage

from mimm import get_model, list_models

# List available models
print(list_models())

# Load a model and convert weights from torchvision
model = get_model('resnet50', pretrained=True)

Models

Classification

• AlexNet
• VGG + BatchNorm variants
• Resnet
• Wide Resnet
• MobileNetV3
• ViT
• Swin Transformer
• Swin v2

Acknowledgements

This repository is based on the following repositories:

• PyTorch Image Models
• Torchvision