这个功能作者没有写出tutorial,所以我写了,不是为了重复,而是补充,作者的教程官网,必须看,必须做笔记记下改什么:
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
首先需要安装caffe,并且能够make runtest成功,没有错误https://github.com/alexgkendall/caffe-segnet,注意如果要用cudnn5.1加速,那就要用作者提供的这个网址https://github.com/TimoSaemann/caffe-segnet-cudnn5。这个caffe特指作者改的,不是官网下的。之后下载作者提供的网络代码https://github.com/alexgkendall/SegNet-Tutorial。
由于我们要用作者提供的训练好的网络参数,或者说权重,所以还需要下载这个:https://github.com/alexgkendall/SegNet-Tutorial/blob/master/Example_Models/segnet_model_zoo.md
建议不要忽视作者写的readme英文以及任何英文。
那么所有格式是caffemodel的都是可以用的。
之后你要改什么呢,当然是1训练语句:你要用什么prototxt,还有2两个.py文件,以及所有3用到的路径。也就是说为了保险,你可以把所有prototxt打开,把绝对路径改成相对路径。另外注意你的4文件层次是不是和作者给的一样,不一样相对路径都要改。同时如果是5cpu跑代码还要注意把.py中的gpu也改成cpu,6.prototxt文件中的GPU也要改成cpu。
再说个关于功能的,如果你想7改变下输出形式你可以改test_segmentation_camvid.py里面的代码,比如你想输出自己的segment结果你就把scipy那行注释去了,然后把参数rgb改成label,命名也要写成‘media\tinaE\+"i"+"_segnet"‘类似这种命名方式。
好了,完。