MMPretrain是一个预训练开源算法框架,目的是提供各种预训练主干网络,并支持不同的预训练策略。包含:主干模型、自监督学习算法、多模态学习算法,支持各种类型数据集
主干模型支持ResNet,Vision Transformer, VGG,MobileNet V2/V3,ShuffleNet v1/v2/v3等。
自监督学习算法支持SimCLR,MAE,MoCo v1/v2/v3
多模态算法支持CLIP,BLIP,OFA
MMPretrain是一个预训练开源算法框架,目的是提供各种预训练主干网络,并支持不同的预训练策略。包含:主干模型、自监督学习算法、多模态学习算法,支持各种类型数据集
主干模型支持ResNet,Vision Transformer, VGG,MobileNet V2/V3,ShuffleNet v1/v2/v3等。
自监督学习算法支持SimCLR,MAE,MoCo v1/v2/v3
多模态算法支持CLIP,BLIP,OFA