Update README.md
This commit is contained in:
23
README.md
23
README.md
@@ -19,8 +19,25 @@
|
|||||||
我们将mouse_data.csv用vscode打开,选择一些数据剪切到mouse_data_test.csv中。
|
我们将mouse_data.csv用vscode打开,选择一些数据剪切到mouse_data_test.csv中。
|
||||||
## 训练模型
|
## 训练模型
|
||||||
运行train.py,程序就会开始训练模型。最终我们能看到控制台打印出的一条test数据。是一个dx,dy拟合出的十个点。
|
运行train.py,程序就会开始训练模型。最终我们能看到控制台打印出的一条test数据。是一个dx,dy拟合出的十个点。
|
||||||
|
<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));//这里的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<float>(0,8,0) << std::endl;
|
||||||
|
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|||||||
Reference in New Issue
Block a user