上接: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, 誰用誰知道!