caffemodel
时间: 2023-12-13 20:31:31 浏览: 193
lenet.caffemodel
4星 · 用户满意度95%
caffemodel是Caffe框架中的一种文件格式,用于存储训练好的模型参数。在Caffe中,模型的结构和参数是分开存储的,其中.prototxt文件存储了模型的结构信息,而.caffemodel文件则存储了模型的参数信息,包括权重和偏置等信息。通过加载.caffemodel文件,我们可以快速地使用已经训练好的模型进行预测或者微调。
如果想要查看caffemodel文件中的权重信息,可以使用Caffe提供的Python接口来快速查看。具体步骤如下:
1.导入caffe和numpy库
```python
import caffe
import numpy as np
```
2.加载模型和参数
```python
deploy = '/home/b622/caffe-action_recog/evaluation/prototxt/pspnet101_VOC2012_deploy.prototxt'
model = '/media/b622/My Passport/pspnet101resultbn/train_iter_100.caffemodel'
net = caffe.Net(deploy, model, caffe.TEST)
```
3.获取指定层的权重信息
```python
w = net.params['conv1_1_3x3_s2/bn'][0].data
print(w)
```
其中,'conv1_1_3x3_s2/bn'是指定的层名称,表示获取该层的权重信息。
阅读全文