yellowdog/mouse_control
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
suixin1424
2024-02-02 17:43:44 +08:00
committed by GitHub
parent a1a6c36502
commit 7ddb7969cc
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -30,12 +30,10 @@ net = cv::dnn::readNetFromONNX("mouse.onnx");
if (!net.empty()) if (!net.empty())
{ {
cv::Mat blob(1, 2, CV_32F, cv::Scalar(100, 100));//这里的100，100是输入dx，dy cv::Mat blob(1, 2, CV_32F, cv::Scalar(100, 100));//这里的100，100是输入dx，dy
int sizes[] = { 1, 9, 2 }; int sizes[] = { 1, 10, 2 };
cv::Mat mat(3, sizes, CV_32F); cv::Mat mat(3, sizes, CV_32F);
std::cout << (mat.cols) << std::endl;
std::cout << blob.size << std::endl;
net.setInput(blob,"input"); net.setInput(blob,"input");
cv::Mat output = net.forward("output");//输出为1*9*2 cv::Mat output = net.forward("output");//输出为1*10*2
std::cout << output.at<float>(0,8,0) << std::endl; std::cout << output.at<float>(0,8,0) << std::endl;
} }