如何获得预训练的VGG16权重的逆矩阵?

时间:2019-05-17 11:27:30

标签: tensorflow

目前,我正在尝试基于预训练的vgg16或vgg19构建卷积自动编码器。我想知道如何得到权重的逆矩阵。

1 个答案:

答案 0 :(得分:0)

我刚刚弄清楚了.. :)希望它对其他人有帮助。下面是代码:

weights = model.get_weights()

needed_weight = weights[index_number]

权重将返回神经网络中所有权重的列表,然后选择您要查找的权重。

仅此而已。