接上篇:
Knight:20191123 使用 Pybind11 和 OpenCV 创建 Python 库zhuanlan.zhihu.com最近花了一段时间, 编写测试通过了使用 Rust 版 Tensorflow 和 OpenCV 进行 OpenPose 推理,并调用 C++ 写的后处理 so 库进行后处理的小 demo。基本流程与 C++ 版 Tensorflow 推理类似:创建 graph,读取 pb 文件,创建 session,设置输入输出,运行 session,获取结果。
其中比较折腾的是前处理和后处理部分,现分享下一个用 Rust 中 Tensorflow Tensor 与 OpenCV Mat 转换的小栗子,供参考:
fn
输出结果为:
data:
献给 @AliceInt-ZLJ by Knight @2019/11/27
(完)