diff --git a/src/AngleNet.cpp b/src/AngleNet.cpp index 0876e57..437fb86 100644 --- a/src/AngleNet.cpp +++ b/src/AngleNet.cpp @@ -82,7 +82,7 @@ Angle AngleNet::getAngle(cv::Mat &src) { std::vector inputNames = {inputNamesPtr.data()->get()}; std::vector 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 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 index 8521a04..a3a5d3a 100644 --- a/src/CrnnNet.cpp +++ b/src/CrnnNet.cpp @@ -122,7 +122,7 @@ TextLine CrnnNet::getTextLine(const cv::Mat &src) { std::vector inputNames = {inputNamesPtr.data()->get()}; std::vector 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 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 index 45e2736..74b93aa 100644 --- 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 inputNames = {inputNamesPtr.data()->get()}; std::vector 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 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 index 272f34c..7c3ea3f 100755 --- a/src/OcrUtils.cpp +++ b/src/OcrUtils.cpp @@ -93,8 +93,8 @@ void drawTextBox(cv::Mat &boxImg, const std::vector &box, int thickne } void drawTextBoxes(cv::Mat &boxImg, std::vector &textBoxes, int thickness) { - for (auto & textBoxe : textBoxes) { - drawTextBox(boxImg, textBoxe.boxPoint, thickness); + for (auto & textBox : textBoxes) { + drawTextBox(boxImg, textBox.boxPoint, thickness); } }