从Squeezenet删除输入张量

时间:2017-11-30 11:52:30

标签: python tensorflow qualcomm

我使用Facenet创建了一个Squeezenet模型 我想使用Qualcomm NPE转换我的Squeezenet张量流模型。问题是,Qualcomm的NPE仅支持3D张量作为输入。 Squeezenet使用图像张量加上phase_train张量(4D张量)作为输入。该模型可在此处找到:squeezenet.py

我的问题是:如何从squeezenet模型中删除phase_train张量以仅输入图像? 也许人们可以在冻结图表时删除phase_train?

我认为两个输入张量都在第46行中定义:

with tf.variable_scope('squeezenet', [images], reuse=reuse):
        with slim.arg_scope([slim.batch_norm, slim.dropout], is_training=phase_train):

0 个答案:

没有答案