使用python接口來運行caffe程序,主要的原因是python非常容易可視化。所以不推薦大家在命令行下面運行python程序。如果非要在命令行下面運行,還不如直接用 c++算了。
推薦使用jupyter notebook,spyder等工具來運行python代碼,這樣才和它的可視化完美結合起來。
因為我是用anaconda來安裝一系列python第三方庫的,所以我使用的是spyder,與matlab界面類似的一款編輯器,在運行過程中,可以查看各變量的值,便于理解,如下圖:

只要安裝了anaconda,運行方式也非常方便,直接在終端輸入spyder命令就可以了。
在caffe的訓練過程中,我們如果想知道某個階段的loss值和accuracy值,并用圖表畫出來,用python接口就對了。

# -*- coding: utf-8 -*-"""Created on Tue Jul 19 16:22:22 2016@author: root"""import matplotlib.pyplot as plt import caffe caffe.set_device(0) caffe.set_mode_gpu() # 使用SGDSolver,即隨機梯度下降算法 solver = caffe.SGDSolver('/home/xxx/mnist/solver.最后生成的圖表在上圖中已經顯示出來了。
新聞熱點
疑難解答