paddleseg飞桨旗下主要用于图像分割的开发套件,配置和使用都很简单方便,本篇基于paddleseg2.3进行学习记录与讲解。
源码地址可以从github上获取:https://github.com/PaddlePaddle/PaddleSeg
,其中包含使用文档,本篇基于使用文档和查阅的资料进行学习记录。
首先paddleseg主要包含以下几个目录:
configs:保存不同网络的配置文件。(主要修改的地方)
docs:文档
paddleseg:PaddleSeg核心代码,包含训练、评估、推理等文件。(核心的文件)
tools:工具脚本
train.py:训练入口文件
eval.py:评估模型文件
predict.py:预测文件
首先来学习一下数据的准备,我们想要使用paddleseg做自己的数据分割,那么首先就需要将数据处理成paddleseg支持的格式。
假设我们已经有了数据和标注数据,将训练集、验证集、测试集都放在一个文件夹里。
(未标注的数据集需要自己进行标注,这里不再讲解,标注数据格式为灰度图,PddleSeg已支持2种标注工具:LabelMe、EISeg交互式分割标注工具。)
PaddleSeg采用单通道的标注图片,每一种像素值代表一种类别,像素标注类别需要从0开始递增,例如