TensorFlow 认可的 vgg-19 model/feature 提取器

TensorFlow endorsed vgg-19 model/feature extractor

我的任务要求我从 vgg-19 网络的 pool5 层提取特征。以前我是使用 Caffe 来做的,然后使用 Tensorflow 处理提取的特征以进行进一步的训练。但现在,我想在顶部添加一个 deconv 层并微调整个网络以进行端到端训练。现在我的梯度是使用 tensroflow 获得的,而我是使用 caffe 从 vgg-19 获得的特征,所以我想很难将这些梯度反向传播到 caffe。所以这就是我在 tensorflow 中寻找 vgg-19 模型的原因,它可以用于从任何层提取特征。我在网上找到了一些项目,但它们并没有得到 Tensorflow 的官方认可。

对于 tensorflow,您可以使用 TF-Slim library 并使用可用的 pretrained models here. You can check their useful slim walk through notebook 如何使用它们。