代码及相关库的下载
- 从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文件