diff --git a/README.md b/README.md index 491c548..7cd6867 100644 --- a/README.md +++ b/README.md @@ -19,8 +19,25 @@ 我们将mouse_data.csv用vscode打开,选择一些数据剪切到mouse_data_test.csv中。 ## 训练模型 运行train.py,程序就会开始训练模型。最终我们能看到控制台打印出的一条test数据。是一个dx,dy拟合出的十个点。 + ## 验证 - 我们将刚刚得到的十个点放到show.py中,观察散点图,发现其轨迹类似于本人鼠标移动轨迹。 -todo -使用onnxruntime在c++上进行推理 + +## c++推理 + +```c++ +cv::dnn::Net net; +net = cv::dnn::readNetFromONNX("mouse.onnx"); +if (!net.empty()) +{ + cv::Mat blob(1, 2, CV_32F, cv::Scalar(100, 100));//这里的100,100是输入dx,dy + int sizes[] = { 1, 9, 2 }; + cv::Mat mat(3, sizes, CV_32F); + std::cout << (mat.cols) << std::endl; + std::cout << blob.size << std::endl; + net.setInput(blob,"input"); + cv::Mat output = net.forward("output");//输出为1*9*2 + std::cout << output.at(0,8,0) << std::endl; + +} +```