Segnet分割网络caffe教程（三）：A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation 图像语义分割

前提：

已经安装好caffe-segnet，并且下载了SegNet-Tutorial-master。

如没有，请参考安装教程：

Caffe-SegNet安装笔记（基于Python2.7）：https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/87256632

 

注意电脑的显存必须是2G，或以上，否则总是显示内存溢出，我的笔记本就是1G的显卡，运行失败，在台式机上才运行成功。
 

下载模型：bayesian_segnet_camvid.caffemodel

模型下载地址：

http://mi.eng.cam.ac.uk/~agk34/resources/SegNet/

或者：https://github.com/alexgkendall/SegNet-Tutorial/blob/master/Example_Models/segnet_model_zoo.md

 

修改 SegNet/CamVid/val.txt 的文件内容，修改为自己的路径：

/home/zqzy/SegNet/CamVid/val/0016E5_07959.png /home/zqzy/SegNet/CamVid/valannot/0016E5_07959.png

运行：

python Scripts/test_bayesian_segnet.py --model Example_Models/bayesian_segnet_camvid.prototxt --weights Example_Models/bayesian_segnet_camvid.caffemodel --colours Scripts/camvid12.png --data /home/zqzy/SegNet/CamVid/val.txt  #這裏的data是 CamVid中的val.txt文件中路徑下的圖片