使用张量流中预先训练的权重进行微调

时间:2018-05-09 19:21:49

标签: python tensorflow

我正在尝试使用预训练网络(VGG19)的卷积层作为特征提取器,并且还能够在训练期间微调最后一层。 我所拥有的是tensorflow提供的vgg19.ckpt: http://download.tensorflow.org/models/vgg_19_2016_08_28.tar.gz (通过https://github.com/tensorflow/models/tree/master/research/slim找到)

  • 我遇到的问题是:我如何只从这个ckpt文件中提取我感兴趣的图层的特定权重?
  • 所以基本上我第一次运行训练时应该从这个ckpt加载我的图形的一些权重并将其余的值初始化为随机值,但是下次它应该从包含新权重的ckpt加载(包括固定重量)。

是否有任何示例说明如何执行此操作? (从ckpt文件加载特定权重,并在训练期间在新图表中对它们进行微调?)

谢谢!

Ĵ

0 个答案:

没有答案