Caffe學習系列(22):caffe圖形化操作工具digits運行實例


上接:Caffe學習系列(21):caffe圖形化操作工具digits的安裝與運行

經過前面的操作,我們就把數據准備好了。

一、訓練一個model

右擊右邊Models模塊的” Images" 按鈕 ,選擇“classification"

在打開頁面右下角可以看到,系統提供了一個caffe model,分別為LeNet, AlexNet, GoogLeNet, 如果使用這三個模型,則所有參數都已經設置好了,就不用再設置了。

在下面,系統為我們列舉出了本機所帶的顯卡,我們可以選擇其中一塊進行運行。

在最下面,輸入一個model name, 就可以點擊create 按鈕了。如果有些選項不對,會有錯誤提示,很人性化。

在訓練過程頁面,左上角顯示了生成的配置文件名稱 (放在job目錄文件下,默認路徑為:/usr/share/digits/digits/jobs/),運行過程中保存的caffemodel快照也保存在這個目錄下面。

頁面中間顯示了訓練和測試的數據信息,右面顯示了訓練所用的時間和gpu使用情況,下面就是一些實時化圖表,可以看到訓練階段的loss, 測試階段的loss和accuracy,相當方便,甚至還可以看到學習率的變化情況,吃驚吧!

 

模型訓練好后,直接就可以在下面進行測試了。

二、測試新來的圖片

將頁面拖到最下面,選擇Upload imager按鈕,加載一幅測試圖片。在 /home/username/mnist/test/ 下面有大量的測試圖片,隨便選一張就可以了。

也可以通過在Image URL方框里,輸入一張網上的圖片地址來進行測試。

加載好測試圖片,在 Show visualizations and statistics 選擇模式框上點上勾。

點擊”Classify One" 按鈕就可以開始測試了。

如果你不是對一張圖片進行測試,而是一個測試集,則是在" Upload Image List"這個地方,選擇測試圖片的列表清單文件(如 val.txt)

系統會彈出一個新的頁面,顯示top-5的分類情況 ,同時digits還提供了測試數據與權值的可視化和統計信息。

 

最后一句話總結,nvidia digits, 誰用誰知道!


免責聲明!

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



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