MMClassification包含多种丰富的模型,支持多种数据集,用于对数据进行分类。
代码仓库:https://github.com/open-mmlab/mmclassification
文档教程:https://mmclassification.readthedocs.io/en/latest/
推理工具使用说明:https://mmclassification.readthedocs.io/en/latest/getting_started.html#inference-asingle-image
训练工具使用说明:
https://mmclassification.readthedocs.io/en/latest/getting_started.html#train-a-model
使用 MIM 工具实现训练和测试:使用说明:https://github.com/open-mmlab/mim
环境搭建:https://mmcv.readthedocs.io/zh_CN/latest/get_started/article.html
深度学习模型的训练涉及几个方面:
− 模型结构 模型有几层、每层多少通道数等等
− 数据集 用什么数据训练模型:数据集划分、数据文件路径、数据增强策略等等
− 训练策略 梯度下降算法、学习率参数、batch_size、训练总轮次、学习率变化策略等等
− 运行时 GPU、分布式环境配置等等
− 一些辅助功能 如打印日志、定时保存checkpoint等等
预训练模型库