MMPretrain环境搭建与配置
安装指定分支的 MMEngine MMEngine new_config 环境搭建
git clone -b new_config_type https://github.com/HAOCHENYE/mmengine.git mmengine_new_config
cd mmengine_new_config
# 如果是新环境,没有安装 MMEngine 相关的依赖,需执行
pip install -e .
算法库适配
基于 new_config 分支,源码编译算法库
源码编译 MMCV
注意:
1. -b new_config 是为了切换支持mmengine的分支
2. 此处的“.[multimodal]”为添加多模态依赖,代表要安装多模态相关的额外依赖,
3. MMCV_WITH_OPS=1 切换到和mmpretrain 配合使用的分支
4. 编译安装mmcv时会很慢,请耐心等待
config文件
resnet18_fintune.py
原始的resnet18_8xb32_in1k.py中__base__中提到的几个文件中的内容复制进去,后面会对这个文件修改或者作为继承文件
resnet18_fintuneM.py
修改复制esnet18_fintune.py模板中所需修改的参数
resnet18_fintuneM2.py 继承机制,简单明了
基于继承机制,继承修改自resnet18_fintune.py
注意:resnet18_fintuneM2.py 与resnet18_fintuneM.py 等价
resnet18_fintuneM.py采用了完整的模板修改,
而resnet18_fintuneM2.py使用继承机制,只呈现了修改的关键参数,更加简单明了
修改要点
num_class=2 #对应数据集种类个数
init_cfg = dict(type='Pretrained', checkpoint='https://download.openmmlab.com/mmclassification/v0/resnet/resnet18_8xb32_in1k_20210831-fbbb1da6.pth')
checkpoint model weight 链接寻找链接和截图
欢迎来到 MMPretrain 中文教程! — MMPretrain 1.0.0rc8 文档
ResNet — MMPretrain 1.0.0rc8 文档
先建个档案 后面慢慢补充。。。