使用Tensorflow训练自己的分割数据

数据准备

首先，需要将准备好的原始数据与其标签文件放在固定目录下，然后建立标签索引文件train.txt与val.txt，具体格式如图所示，左侧是原图，右侧是标签图：

然后建立test.txt文件，具体如图所示：

设计网络结构

比如使用FCN的机制来进行分割，以下面的网络结构为例：

修改配置文件

比如本次训练使用高速的数据，将训练配置文件hypes/KittiSeg_highspeed.json中的data部分修改为自己数据的路径文件，jitter部分修改输入图像尺寸为自己图像尺寸，分别如下两图所示：

开始训练

使用如下命令来进行训练：
python train.py –hypes hypes/KittiSeg_highspeed.json
请千万注意，这里的–hypes是两个“-”！！！