如何在tensorflow中提取图层或ValueError:Shape(?,4096)必须具有等级1

时间:2016-10-06 07:24:49

标签: python machine-learning neural-network tensorflow

操作系统:CentOS 6 - 流浪汉官方包装盒。没有安装CUDA。

安装: 1.安装Anaconda 4.0.9; 2. Tensorflow版本:0.10.0rc0; 3. Python 3.5,numpy 1.11.2,scipy 0.18.1;

重现

标准的VGG型号有16层。我尝试使用以下代码提取vgg.fc2图层:http://pastebin.com/A1qd7g2e并出现错误(下面的堆栈跟踪)。调用session.run后,我也收到此错误。我是否以正确的方式尝试提取图层?

Stacktrace - http://pastebin.com/eaQi1T0R

P.S。我是tenserflow的新用户,不知道任何尝试过的解决方案。我发现了这个问题 - https://github.com/tensorflow/tensorflow/issues/3815,但它已关闭,我打开了另一个问题 - https://github.com/tensorflow/tensorflow/issues/4773,任何想法都会有所帮助。

1 个答案:

答案 0 :(得分:0)

原因是我们只能在sess.run中运行图形后调用图形节点vgg.fc2。有关详细信息,请参阅https://github.com/tensorflow/tensorflow/issues/4773