c++ pytorch部署相关操作代码 libtorch
toTensor() c++ pytorch部署相关操作代码
以下笔记情况:
pytorch模型转成c++后进行识别(其实是我是做单字分类)所写的主要代码。
头文件
#include <torch/script.h>
#include <iostream>
#include <memory>
#include <string>
#include <cmath>
// opencv 头文件
#include <opencv2/highgui/
原创
2021-05-08 15:53:07 ·
657 阅读 ·
6 评论