1. 程式人生 > >利用segnet作者訓練好的網絡測試自己的圖

利用segnet作者訓練好的網絡測試自己的圖

由於 cts blog 語句 自己 log 還需 錯誤 還要

這個功能作者沒有寫出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"‘類似這種命名方式。

好了,完。

利用segnet作者訓練好的網絡測試自己的圖