Question

我正在尝试将base64字符串数据解码为张量，以馈入模型进行预测。 tf python中有一个函数（tf.image.decode_image）用于将字符串转换为张量，但在tf c ++中找不到任何api。如果没有可用的api，如何解决此问题？

Answer 1

我认为您要先DecodeBase64，然后再DecodeRaw。

#include <vector>
#include "tensorflow/cc/client/client_session.h"
#include "tensorflow/cc/ops/standard_ops.h"
#include "tensorflow/cc/ops/string_ops.h"
#include "tensorflow/cc/ops/parsing_ops.h"
#include "tensorflow/core/framework/tensor.h"

int main() {
  using namespace tensorflow;
  using namespace tensorflow::ops;
  Scope root = Scope::NewRootScope();
  // Float32 array [1. 2. 3. 4. 5. 6.] base64 encoded
  auto b64 = Const(root, "AACAPwAAAEAAAEBAAACAQAAAoEAAAMBA");
  // Decode base64
  auto decoded = DecodeBase64(root, b64);
  // Parse bytes
  auto parsed = DecodeRaw(root, decoded, DT_FLOAT32);
  // Run
  std::vector<Tensor> outputs;
  ClientSession session(root);
  // Get parsed data
  TF_CHECK_OK(session.Run({parsed}, &outputs));
  // outputs[0] == [1. 2. 3. 4. 5. 6.]
  LOG(INFO) << outputs[0].flat<float>();
  return 0;
}

在tensorflow c ++中，是否有将base64字符串解码为张量的函数？

1 个答案: