修复拼写错误等

RapidAI · Oct 28, 2022 · 037bf4c · 037bf4c
1 parent 1340d65
commit 037bf4c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/src/AngleNet.cpp b/src/AngleNet.cpp
@@ -82,7 +82,7 @@ Angle AngleNet::getAngle(cv::Mat &src) {
     std::vector<const char *> inputNames = {inputNamesPtr.data()->get()};
     std::vector<const char *> outputNames = {outputNamesPtr.data()->get()};
     auto outputTensor = session->Run(Ort::RunOptions{nullptr}, inputNames.data(), &inputTensor,
-                                     inputNamesPtr.size(), outputNames.data(), outputNamesPtr.size());
+                                     inputNames.size(), outputNames.data(), outputNames.size());
     assert(outputTensor.size() == 1 && outputTensor.front().IsTensor());
     std::vector<int64_t> outputShape = outputTensor[0].GetTensorTypeAndShapeInfo().GetShape();
     int64_t outputCount = std::accumulate(outputShape.begin(), outputShape.end(), 1,

diff --git a/src/CrnnNet.cpp b/src/CrnnNet.cpp
@@ -122,7 +122,7 @@ TextLine CrnnNet::getTextLine(const cv::Mat &src) {
     std::vector<const char *> inputNames = {inputNamesPtr.data()->get()};
     std::vector<const char *> outputNames = {outputNamesPtr.data()->get()};
     auto outputTensor = session->Run(Ort::RunOptions{nullptr}, inputNames.data(), &inputTensor,
-                                     inputNamesPtr.size(), outputNames.data(), outputNamesPtr.size());
+                                     inputNames.size(), outputNames.data(), outputNames.size());
     assert(outputTensor.size() == 1 && outputTensor.front().IsTensor());
     std::vector<int64_t> outputShape = outputTensor[0].GetTensorTypeAndShapeInfo().GetShape();
     int64_t outputCount = std::accumulate(outputShape.begin(), outputShape.end(), 1,

diff --git a/src/DbNet.cpp b/src/DbNet.cpp
@@ -131,7 +131,7 @@ DbNet::getTextBoxes(cv::Mat &src, ScaleParam &s, float boxScoreThresh, float box
     std::vector<const char *> inputNames = {inputNamesPtr.data()->get()};
     std::vector<const char *> outputNames = {outputNamesPtr.data()->get()};
     auto outputTensor = session->Run(Ort::RunOptions{nullptr}, inputNames.data(), &inputTensor,
-                                     inputNames.size(), outputNames.data(), 1);
+                                     inputNames.size(), outputNames.data(), outputNames.size());
     assert(outputTensor.size() == 1 && outputTensor.front().IsTensor());
     std::vector<int64_t> outputShape = outputTensor[0].GetTensorTypeAndShapeInfo().GetShape();
     int64_t outputCount = std::accumulate(outputShape.begin(), outputShape.end(), 1,

diff --git a/src/OcrUtils.cpp b/src/OcrUtils.cpp
@@ -93,8 +93,8 @@ void drawTextBox(cv::Mat &boxImg, const std::vector<cv::Point> &box, int thickne
 }
 
 void drawTextBoxes(cv::Mat &boxImg, std::vector<TextBox> &textBoxes, int thickness) {
-    for (auto & textBoxe : textBoxes) {
-        drawTextBox(boxImg, textBoxe.boxPoint, thickness);
+    for (auto & textBox : textBoxes) {
+        drawTextBox(boxImg, textBox.boxPoint, thickness);
     }
 }