本文总结了SegNet-tensorflow的编译过程，包括从github下载代码、安装依赖库，特别是解决tensorflow版本兼容问题。同时，详细介绍了数据集CamVid的下载及放置，并提供了训练前的配置步骤，如创建Logs文件夹、修改配置文件路径。在训练过程中，强调了路径正确性对避免OutOfRangeError错误的重要性，以及处理GPU设备名问题的建议。
代码及相关库的下载
- 从github上下载SegNet-tensorflow的代码，命令如下：
$git clone https://github.com/tkuanlun350/Tensorflow-SegNet.git SegNet
- 安装readme中要求的库。
$pip install tensorflow-gpu==1.10
$pip install pillow
$pip install scikit-image
scikit-image 需要其他库如matplotlib等，如果装的时候遇到问题，按提示信息把依赖的库都装上就行。
注意：这里的tensorflow版本一定要适配，不然会出现loaded runtime CuDNN library：7.1.4 but source was compiled with 7.2.1 这样版本不兼容的问题。
3. 下载训练用数据集CamVid
下载地址：https://github.com/alexgkendall/SegNet-Tutorial
下载其中的Camvid文件夹，然后放在SegNet目录下。
编译SegNet-tensorflow代码
- 进入下载的SegNet的代码根目录下，新建一个Logs的文件夹用来存放日志。
$mkdir Logs
-
用 gedit 打开
最低0.47元/天 解锁文章
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
