tensorflow c++资料

最新推荐文章于 2024-10-10 16:37:17 发布

原创最新推荐文章于 2024-10-10 16:37:17 发布 · 517 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

tensorflow

当前文章被收录于：

tensorflow

2 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了如何在C++环境中将OpenCV的Mat对象转换为TensorFlow中的Tensor对象，包括创建Tensor、设置形状及数据类型等步骤，并解决了由于protobuf版本不一致导致的问题。

tensorflow-object-detection-cpp
Direct access to Tensor Buffers in C++ interface #8033

Mat转Tensor

img = cv::imread(img_path);
TensorShape shape({1, img.rows , img.cols, 3});
input_tensor = Tensor(tensorflow::DT_UINT8, shape);
uint8_t *p = input_tensor.flat<uint8_t>().data();
Mat fakeMat(img.rows, img.cols, cv::CV_8UC3, p);
img.convertTo(fakeMat, cv::CV_8UC3);