热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

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

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

数据准备

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

这里写图片描述

然后建立test.txt文件,具体如图所示:
这里写图片描述

设计网络结构

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

这里写图片描述

修改配置文件

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

这里写图片描述

这里写图片描述

开始训练

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


推荐阅读
author-avatar
可可伦-惊叹号
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有