qq_36416259的博客

原创 Tensorflow:如何加载神经网络部分参数

需求如下：先用Resnet50在ImageNet上预训练，最后一层输出为类别数量，设为1000。然后将保存下来的参数迁移到PascalVoc上训练。问题：由于PascalVoc只有20类，所以Resnet50最后一层输出要改为20。此时直接用tf.train.Saver()的restore，因为预训练的参数最后一层resnet50/fc长度为1000，而新模型最后一层resnet50/fc长度为2...