3

我使用Facenet创建了一个 Squeezenet 模型 我想使用 Qualcomm NPE 转换我的 Squeezenet tensorflow 模型。问题是,高通的 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):
4

0 回答 0