参考http://www.voidcn.com/blog/zackzhaoyang/article/p-5753454.html
1、读取参数
net.params('conv1', 1).get_data
2、读取feature map
net.blobs('conv1').get_data
3、设置模式
caffe.set_mode_gpu()
4、设置显卡
caffe.set_device(0)
参考http://www.voidcn.com/blog/zackzhaoyang/article/p-5753454.html
1、读取参数
net.params('conv1', 1).get_data
2、读取feature map
net.blobs('conv1').get_data
3、设置模式
caffe.set_mode_gpu()
4、设置显卡
caffe.set_device(0)