mmit 0.0.2

A Python package for computer vision experiments and research.

Multi-Models for Images in pyTorch (MMIT)

mmit is a python library with pretrained building blocks for Computer Vision models.

Why mmit?

mmit is a library that provides building blocks for Computer Vision models. Heavily inspired by SMP, we aim at providing a unified interface for all encoder and decoders, so that any composition is plug and play. It is designed to be flexible in order to support rapid implementation and evaluation of novel research. It is also designed to be fast, modular, and extensible. mmit is built on top of PyTorch.

To Do List

In the future, we plan to add support for:

• timm encoders
• some of timm transformers encoders
• torchvision / torchub models
• UNet and UNet++ decoders
• other decoders
• multiple heads
• popular loss function
• popular datasets
• popular metrics

Awesome Sources

• PyTorch
• PyTorch Lightning
• Timm
• SMP