本次实战经验主要为Config文件的撰写。文件的撰写基本可以分为以下几个步骤:
- 选择继承的文件,一般选择成熟的文件对应的config即可
- 选择数据集,如果要使用自己的数据集,那么要设置相应的模块。这些模块包含:
- 数据集位置
- 数据集元信息
- 数据集标注文件
- 选择模型,如果对数据集做出了更改,那么就需要对模型也做出修改。本次使用的三角板数据集与耳朵数据集都只有一类。
- 配置训练流程、测试流程。这一步踩坑比较大。由于提供的数据集图像质量高,导致处理缓慢,需要配置相对应的配置才能提高性能。
今天的课程主要分为三个部分:安装MMDet和MMPose、训练MMDet模型、训练MMPose模型。
-
安装MMDet和MMPose 按照教程在jupyter notebook中一步步走就可以,如果项目不能直接git clone,可以本地下载后上传,注意下载对应分支。
-
训练MMDet模型 作业的ear放到/data目录下,数据集也放到/data目录下,样例是可以直接运行的。