python - 我如何知道 "bias"是否存在于图层中？

Question

我正在尝试使用 pycaffe 读取 caffe 网络中的权重和偏差。这是我的代码

weight = net.params[layer_name][0].data
bias = net.params[layer_name][1].data

但是，我的网络中的某些层没有偏差，因此会出现索引超出范围的错误。

所以我的问题是我可以使用

if(net.params[layer_name][1] exists):
    bias = net.params[layer_name][1].data

控制对偏差的分配？那么代码怎么写呢？

Answer 1

您可以简单地迭代net.params[layer_name]:

layer_params = [blob.data for blob in net.params[layer_name]]

这样，您就可以获得所有layer_params(对于某些层来说可能超过2，例如"BatchNorm")

如果只想检查第二个参数blob，可以使用len:

if len(net.params[layer_name]) >= 2:
    bias = net.params[layer_name][1].data

<小时/>

附注，
可能的情况是 net.params[layer_name] 并不完全是一个 python list，而是一些 python boost 包装对象，因此您可能需要将其显式转换为list (list(net.params[layer_name])) 在我在此答案中建议的一些方法中。