您的位置:首页 > 其它

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

2017-04-11 10:26 260 查看

数据准备

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



然后建立test.txt文件,具体如图所示:



设计网络结构

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



修改配置文件

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





开始训练

使用如下命令来进行训练:

python train.py –hypes hypes/KittiSeg_highspeed.json

请千万注意,这里的–hypes是两个“-”!!!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: