本节内容: 语义分割代码实践课 MMSegmentation
实现语义分割的两个层面:
- 调用模型库中已有的模型去对一张新的图片、新的视频去进行语义分割预测
- 标注自己的语义分割数据集,训练自己的模型,解决实际问题
语义分割在MMSegmentation,实例分割在MMDetection(因为要先进行目标检测)。
语义分割目标是对每一个像素进行分类。
实践作业内容:
1、数据集标注(可选)
使用Labelme、LabelU等数据标注工具,标注多类别语义分割数据集,并保存为指定的格式。
2、数据集整理
划分训练集、测试集
3、使用MMSegmentation训练语义分割模型
在MMSegmentation中,指定预训练模型,配置config文件,修改类别数、学习率。
4、用训练得到的模型预测
获得测试集图片或新图片的语义分割预测结果,对结果进行可视化和后处理。
5、在测试集上评估算法的速度和精度性能
6、使用MMDeploy部署语义分割模型(可选)
本课教程代码:
https://github.com/TommyZihao/MMSegmentation_Tutorials/tree/main/20230206