图像分割之流程总结
1.前言
根据图像分割的方法,这篇文章主要就是做一个简单的总结。整体的步骤和目标识别的步骤差不多,主要就是数据处理、训练和预测,最后就是模型部署。因为前面我尝试过用yolo进行了操作,虽然能完整的复现整个过程,但是没有能实际拿的出手的,后来我了解到,使用DeepLab可能效果更好,于是就准备用这个进行测试和训练,看看最终的结果如何。
2.数据处理
(1)矢量标注
创建shp文件,勾绘图形,用arcpro对不同地物进行勾绘,生成标注矢量图层。
(2)赋值属性
根据地物属性,设置id,不同地物类型设置不同的id。
(4)数据质检
对生产的矢量数据进行拓扑检查,处理压盖、空洞问题,数据中不能出现模糊的定义,数据分类越精确,识别效果越好,未知类型赋值为背景id。
(5)格式转换
将shp文件按照id号属性,转为栅格图像,生成Mask图像,
(6)数据对齐
将原始影像和Mask图像用进行校准,用同一个矩形进行裁剪,保证数据范围一致。
(7)数据分割
利用分割代码,将原始影像和Mask影像进行分割,生成 512x512 的切片,并按照 7:3 的比例分成训练集和测试集,供模型训练和测试。
【1】.【实战指南】3步极速上手:用SegFormer实现遥感图像语义分割 SegFormer作为专为语义分割设计的Transformer模型,通过创新的分层编码器和轻量级解码器架构,在保持高精度的同时大幅降低了计算成本,特别适合处理大尺寸遥感图像
【2】.语义分割笔记(一):基于PaddleSeg使用Transfomer模型对航空遥感图像分割
