问题描述
在Python代码中，将图像数据分配给张量image_batch:
In the Python code, the image data is assigned to tensor image_batch:
部分代码:
image_data = misc.imread(image_path)
image_batch = graph.get_tensor_by_name("input:0")
phase_train_placeholder = graph.get_tensor_by_name("phase_train:0")
embeddings = graph.get_tensor_by_name("embeddings:0")
feed_dict = {image_batch: np.expand_dims(image_data, 0), phase_train_placeholder: False}
rep = sess.run(embeddings, feed_dict=feed_dict)
C++ 代码:
const float * source_data = (float*) image.data;
Tensor image_batch(DT_FLOAT, TensorShape({1, 160, 160, 3}));
auto input = image_batch.tensor<float, 4>();
for (int y = 0; y < height; ++y) {
const float* source_row = source_data + (y * width * depth);
for (int x = 0; x < width; ++x) {
const float* source_pixel = source_row + (x * depth);
for (int c = 0; c < depth; ++c) {
const float* source_value = source_pixel + c;
//std::cout << *source_value << std::endl;
input(0, y, x, c) = *source_value;
}
}
}
Tensor phase_train(DT_BOOL, TensorShape());
phase_train.scalar<bool>()() = false;
std::vector<std::pair<string, tensorflow::Tensor>> inputs = {
{ "input:0", image_batch },
{ "phase_train:0", phase_train },
};
std::vector<Tensor> outputs;
Status run_status = session->Run(inputs, {"embeddings:0"}, {}, &outputs);
if (!status.ok()) {
std::cout << status.ToString() << "\n";
return 1;
}
auto output_c = outputs[0].scalar<float>(); //Error here
std::cerr << "SHOW\n";
// Print the results
std::cout << outputs[0].DebugString() << "\n";
std::cout << output_c() << "\n"; // 30
错误:
F tensorflow/core/framework/tensor.cc:493] 检查失败:1 == NumElements() (1 vs. 1792)Must have a one element tensor
F tensorflow/core/framework/tensor.cc:493] Check failed: 1 == NumElements() (1 vs. 1792)Must have a one element tensor
进程结束，退出代码 6
Process finished with exit code 6
推荐答案
在此之前，我犯了一个愚蠢的错误，没有找到关键的错误代码.
Before that, I made a stupid mistake and did not find the key error code.
auto output_c = outputs[0].scalar<float>();
替换:
auto output_c = outputs[0].flat<float>();
所有问题都解决了.
https://github.com/tensorflow/tensorflow/issues/3362(提示我检查什么代码)
https://github.com/tensorflow/tensorflow/issues/3362 (prompted me to check what code)
@Aziuth 谢谢.
@Aziuth Thanks.
这篇关于检查失败:1 == NumElements() (1 vs. 1792)在 Tensorflow C++ 中必须有一个单元素张量的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！