在python中使用caffe,我想访问我的原型文件中指定的所有参数。
使用以下原型文字:
name: "VGG_ILSVRC_16_layers"
layer {
name: "input"
type: "Input"
top: "data"
input_param {
shape {
dim: 10
dim: 3
dim: 224
dim: 224
}
}
}
layer {
name: "conv1_1"
type: "Convolution"
bottom: "data"
top: "conv1_1"
convolution_param {
num_output: 64
pad: 1
kernel_size: 3
}
}
...
我能做到:
>>> import sys, os
>>> import argparse
>>> import caffe
>>> caffenet = caffe.Net(str(deploy), str(cmodel), caffe.TEST)
>>> print caffenet.layers[1].type
Convolution
>>> print caffenet.layers[0].type
Input
但尝试访问其他参数失败:
>>> print caffenet.layers[0].top
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: 'Layer' object has no attribute 'top'
>>> print caffenet.layers[0].input_param.shape
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: 'Layer' object has no attribute 'input_param'
我应该如何访问第一层input_param.shape的前2个维度(10和3),或者conv1_1 num_output参数?