如何在iOS中的tensorflow中传递额外的输入参数?

时间:2017-02-14 11:15:10

标签: ios opencv tensorflow

我将参数发送到tensorflow,如下所示:

 std::string input_layer = "input_node";
    std::string output_layer = "output_node";
    std::string dropout_layer = "keep_prob"; //--- extra parameter

    Status run_status = session->Run({{input_layer, resized_tensor}},
                                       {output_layer}, {}, &outputs);
      if (!run_status.ok()) {
        LOG(ERROR) << "Running model failed: " << run_status;
        return -1;
      }

我试过

 tensorflow::Status run_status = session->Run({{input_layer,dropout_layer, image_tensor}},{output_layer},{}, &outputs);

但它不接受额外的参数 如何发送额外的输入参数?

0 个答案:

没有答案