1. 程式人生 > >caffe實戰記:caffe中python的介面函式

caffe實戰記:caffe中python的介面函式

  • Caffe提供了python的介面(pycaffe),詳見caffe/python資料夾。在python程式碼中import caffe,可以load models(匯入模型),forward and backward(前向、反向迭代), handle IO(資料輸入輸出),visualize networks(繪製net),instrument model solving(自定義優化方法)。所有的模型資料、計算引數都暴露在外,可供讀寫的。

    • caffe.Net是主要介面,負責匯入資料、校驗資料、計算模型的。
    • caffe.Classsifier用於影象分類。
    • caffe.Detector用於影象檢測。
    • caffe.SGDSolver暴露在外的solver的介面。
    • caffe.io處理輸入輸出,資料預處理。
    • caffe.draw視覺化net的結構。
    • Caffe blobs以numpy ndarrays的形式表示,更加方便、高效。