当前位置: 首页 > 工具软件 > PaddleSeg > 使用案例 >

paddleseg学习记录(1)

邹英发
2023-12-01

1总体介绍

paddleseg飞桨旗下主要用于图像分割的开发套件,配置和使用都很简单方便,本篇基于paddleseg2.3进行学习记录与讲解。
源码地址可以从github上获取:https://github.com/PaddlePaddle/PaddleSeg,其中包含使用文档,本篇基于使用文档和查阅的资料进行学习记录。
首先paddleseg主要包含以下几个目录:
configs:保存不同网络的配置文件。(主要修改的地方)
docs:文档
paddleseg:PaddleSeg核心代码,包含训练、评估、推理等文件。(核心的文件)
tools:工具脚本
train.py:训练入口文件
eval.py:评估模型文件
predict.py:预测文件

首先来学习一下数据的准备,我们想要使用paddleseg做自己的数据分割,那么首先就需要将数据处理成paddleseg支持的格式。

2 准备数据

假设我们已经有了数据和标注数据,将训练集、验证集、测试集都放在一个文件夹里。
(未标注的数据集需要自己进行标注,这里不再讲解,标注数据格式为灰度图,PddleSeg已支持2种标注工具:LabelMe、EISeg交互式分割标注工具。)

2.1 灰度图转为伪彩色图(如果不需要转换,直接看2.2)

PaddleSeg采用单通道的标注图片,每一种像素值代表一种类别,像素标注类别需要从0开始递增,例如
 类似资料: