今天介绍了OpenMMLab的MMClassification,第一个库。
模型包括了从VGG到ConvNeXt的很多CNN,轻量化CNN MobileNet,以及Transformer;还有自带的预训练模型和训练策略。和我们平时使用库一样,用Python API即可调用,唯一需要的工作是对应不同场景和需要,修改对应模型的config文件就好。
库中具体的函数使用可以参考Getting Started — MMClassification 0.25.0 documentation。
同时,由于OpenMMLab是基于PyTorch的,不仅有CV部分和基于CV搭建了各种功能的库,也有MIM(openmim)用来管理所有工具。
使用MMCL:
使用MIM:
(trivial process,之后有时间的话再仔细补充)