TensorFlow认可了vgg-19型号/功能提取器

时间:2017-08-02 17:37:23

标签: tensorflow deep-learning caffe autodiff vgg-net

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

1 个答案:

答案 0 :(得分:0)

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