Openmmlab寒假训练营(三)

Openmmlab寒假训练营(三)

MMClassification学习记录

1.tools模块

位于 mmclassification/tools/
功能 包含训练及推理等封装命令

2.configs模块

位于 mmclassification/configs
功能 存放各种模型的训练和预测脚本，每个脚本包含dataset,model,schedule,runtime_setting

3.datasets模块

位置 mmclassification/datasets
功能 存放模型记载数据使用的多种数据集格式的dataloadedr

4.models模块

位置 mmclassification/mmcls/models
功能 存放封装好的各种backbones，损失等供调用

5.schedules模块

位置 mmclassification/configs/base/schedules
功能 定义模型的优化器、学习率和训练过程等参数

6.runtime_settings模块

位置 mmclassification/configs/base/default_runtime.py
功能 定义模型的运行设置

7.数据集的格式

项目	作用
train val test	三个文件夹目录都存放在data父级目录下，这三个目录由data目录和ann.txt组成
…/…/data目录	存放图片数据
…/…/ann.txt	这个可省略,记录图片数据对应的注释，即文件名以及对应类比额index(从0~cnt-1)编号

其他详情信息可参考 MMClassification学习总结