MMPretrain
MMPretrain是一个全新升级的开源算法框架,其继承自MMClassification和MMSelfSup,并开发了许多令人兴奋的新功能。目前,预训练阶段对于视觉识别至关重要。为了改进下游任务,这个模型里有很多优秀的友好的预训练模型。
这就是MMPretrain里支持的两个模型的总览图,根据这两张图片不难看出,这两个分别是KaiMing大神的方法MAE和OpenAI的工作CLIP。
工具箱
支持方法一览,对于预训练模型来说,一些大型的方法应该是需要的。
视觉backbone选讲
-
ResNet
-
ViT
自监督学习支持
-
SimCLR
-
MAE