Update README.md

This commit is contained in:
suixin1424
2024-01-28 20:43:44 +08:00
committed by GitHub
parent f525cf3039
commit e0bc413150

View File

@@ -19,8 +19,25 @@
我们将mouse_data.csv用vscode打开选择一些数据剪切到mouse_data_test.csv中。 我们将mouse_data.csv用vscode打开选择一些数据剪切到mouse_data_test.csv中。
## 训练模型 ## 训练模型
运行train.py程序就会开始训练模型。最终我们能看到控制台打印出的一条test数据。是一个dxdy拟合出的十个点。 运行train.py程序就会开始训练模型。最终我们能看到控制台打印出的一条test数据。是一个dxdy拟合出的十个点。
<img src="./imgs/collect.png">
## 验证 ## 验证
<img src="./imgs/Figure_2.png">
我们将刚刚得到的十个点放到show.py中观察散点图发现其轨迹类似于本人鼠标移动轨迹。 我们将刚刚得到的十个点放到show.py中观察散点图发现其轨迹类似于本人鼠标移动轨迹。
todo <img src="./imgs/Figure_2.png">
使用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));//这里的100100是输入dxdy
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<float>(0,8,0) << std::endl;
}
```