#200 · Primary category: Computer Vision

mmpretrain

beit clip constrastive-learning convnext deep-learning image-classification mae masked-image-modeling mobilenet moco multimodal pretrained-models pytorch resnet self-supervised-learning swin-transformer vision-transformer

OpenMMLab Pre-training Toolbox and Benchmark

Project last updated:11/01/24

GitHub Stars

3.9K

Forks

1.1K

Contributors

129

License

Apache-2.0

Why we included this project

Building a vision model often means choosing between hand-rolled training loops and a toolbox with the pipelines already worked out, and MMPretrain lands firmly on the practical side. It ships a large catalog of pretrained backbones including ResNet, ConvNeXt, Swin Transformer, ViT, and MobileNet, so you can start from a solid weight set instead of training from scratch. Beyond supervised classification, it also covers self-supervised methods like MAE, BEiT, and MoCo, plus multimodal approaches such as CLIP, BLIP, and LLaVA. That breadth matters if you are doing transfer learning or image-text retrieval and want reproducible baselines without wiring separate implementations together. Since it is built on PyTorch and part of the wider OpenMMLab ecosystem, pretrained weights, configs, and task templates tend to drop in cleanly for teams already using those tools.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category