首页 文章

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

提问于
浏览
0

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

1 回答

  • 0

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

相关问题