在Caffe中加載預訓練的模型有兩種常用的方法:
./build/tools/caffe train -solver=path/to/solver.prototxt -weights=path/to/pretrained_model.caffemodel
其中,path/to/solver.prototxt
是你定義的solver配置文件的路徑,path/to/pretrained_model.caffemodel
是預訓練模型的路徑。
import caffe
# 設置Caffe的配置文件和預訓練模型的路徑
model_def = 'path/to/deploy.prototxt'
model_weights = 'path/to/pretrained_model.caffemodel'
# 加載預訓練模型
net = caffe.Net(model_def, model_weights, caffe.TEST)
其中,path/to/deploy.prototxt
是你定義的網絡結構文件的路徑,path/to/pretrained_model.caffemodel
是預訓練模型的路徑。加載成功后,你可以通過net.params
和net.blobs
來訪問網絡參數和中間層數據。