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


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

好了,完。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM