获取Mobilnet v1 Tensorflow的完全连接层

时间:2018-03-27 20:03:39

标签: python tensorflow

伙计们,我正在使用tensorflow Mobilenet v1,我重新训练了最后一层输出4个类,它可以工作。现在,我试图使用最后一个完全连接的层(或最后一个特征铺层而不是softmax层),然后将其提供给LSTM网络。我只是不知道哪个是我需要的最后一层来自mobilenet v1。这是我正在使用的模型:mobilenet_v1 这是我用来查看所有图层的代码,但我不知道选择哪一个:

sed -i '.bak' 's#/home/new/abc/xyz#/arun/old#' test.txt

1 个答案:

答案 0 :(得分:0)

似乎软件之前的最后一个Bottleneck层是我需要使用的。该层返回1x1001矩阵

相关问题