Caffe学习日记8

来源:互联网 发布:罗技淘宝旗舰店有两个 编辑:程序博客网 时间:2024/06/11 02:44

1. 设置Caffe的运行模式

解析:

(1)caffe.set_mode_cpu();

(2)caffe.set_mode_gpu();


2. 初始化网络结构

解析:

net = caffe.Net('D:\caffe-master\models\bvlc_alexnet\deploy.prototxt', 'G:\bvlc_alexnet.caffemodel', 'test');


3. 显示网络Blob和Layer名字

解析:

(1)net.layer_names

(2)net.blob_names


4. 获得第一个卷积层

解析:

conv1_layer = net.layer_vec(2);

说明:class(conv1_layer)为1*1 caffe.Layer数据类型。 


5. 获得第一个卷积层的Blob

解析:

blob1 = conv1_layer.params(1);


6. 获得第一个卷积层的Diff 

解析:

diff1 = conv1_layer.params(2);


7. 获得第一个卷积层Blob的值

解析:

w = blob1.get_data();

说明:class(w)是第一个卷积层Blob的大小,四维数组。


8. 获得第一个卷积层Diff的值

解析:

d = diff1.get_diff();

说明:class(d)是第一个卷积层Diff的大小,四维数组。  

1 0