我使用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):