TensorFlow 支持 vgg-19 模型/特征提取器

Question

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

Answer 1

对于tensorflow，您可以使用TF-Slim library 并使用pretrained models 可用的here。您可以查看他们有用的slim walk through notebook 以了解如何使用它们。