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的显卡，运行失败，在台式机上才运行成功。
 

下载模型：segnet_basic_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/Models/segnet_inference.prototxt 第8行：

source: "/home/zqzy/SegNet/CamVid/test.txt"	# Change this to the absolute path to your data file

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

/home/zqzy/SegNet/CamVid/test/0001TP_008550.png /home/zqzy/SegNet/CamVid/testannot/0001TP_008550.png

运行：

python SegNet/Scripts/test_segmentation_camvid.py --model SegNet/Models/segnet_inference.prototxt --weights SegNet/Example_Models/segnet_basic_camvid.caffemodel --iter 233 #迭代次数可以小一些

 

参考：https://blog.csdn.net/caicai2526/article/details/77169871