如何使用tf.layers在预训练模型中输出层之前插入层

Question

我已经创建了一个CNN，该CNN具有2个Conv层，然后是FC层和输出层(仅2个节点(，然后我使用saver.save(sess，path(保存了参数，现在我想加载预训练的权重并冷冻所有层，然后在FC1之后和输出层之前插入另一个FC2层。

Answer 1

我找到了有用的溶液

convWeights = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES,scope="THE SCOPE OR VARIABLE OR LAYER NAME")                               
convDict = dict([(var.op.name, var) for var in convWeights])
saver = tf.train.Saver(convDict)
with tf.Session() as sess:
    sess.run(init);
    saver.restore(sess, "MODEL.ckpt")