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