Tensorflow android演示使用带有多输入的自定义模型

时间:2017-06-22 07:58:24

标签: android tensorflow

我有一个预训练的模型,在训练阶段使用batch_norm,当我尝试将此模型用于我的android时,它将输出作为固定值返回。

所以我尝试为我的模型做两个输入:&#34; images&#34;和&#34; phase_train&#34; :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;

0 个答案:

没有答案