我有一个预训练的模型,在训练阶段使用batch_norm
,当我尝试将此模型用于我的android时,它将输出作为固定值返回。
所以我尝试为我的模型做两个输入:" images
"和" phase_train
" :images
是图片的向量,phase_train
设置为False
无法batch_norm
,那么如何在Android上为我的模型设置多输入?< / p>
public static Classifier create(
AssetManager assetManager,
String modelFilename,
String labelFilename,
int numClasses,
int inputSize,
int inputSize1,
String inputName,
String outputName)
throws IOException {
TensorFlowImageClassifier c = new TensorFlowImageClassifier();
c.inputName = inputName; # the input is seted here
c.outputName = outputName;