如何从 openvino 模型中获取特定的层权重？

【问题标题】：How can I get a specific layer weights from openvino model?如何从 openvino 模型中获取特定的层权重？
【发布时间】：2022-07-04 13:47:07
【问题描述】：

我有一个经过训练的模型，我想获得例如卷积层权重矩阵。

net = ie.read_network(model='model.xml', weights='model.bin')
ie.set_config(config={"CPU_THREADS_NUM": str(4)}, device_name="CPU")
exec_net = ie.load_network(network=net, device_name="CPU", num_requests=1)

权重存储在哪里？

【问题讨论】：

标签： python deep-learning openvino

【解决方案1】：

推荐使用nGraph函数的get_ordered_ops()方法获取其图节点的拓扑排序列表。

import ngraph as ng
function = ng.function_from_cnn(net)
for node in function.get_ordered_ops():
      print('Node name: {:15} Friendly name: {:10} Op: {:10} {}'.format(node.name, node.get_friendly_name(), node.get_type_name(), node.shape))

【讨论】：

更多示例可以参考nGraph的Python API。