forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix error

This commit is contained in:
LDOUBLEV 2021-05-26 06:55:34 +00:00
parent 2ab0ba91e4
commit 947470a3b2
1 changed files with 80 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
deploy/cpp_infer/src/ocr_rec.cpp
View File

@ -89,8 +89,9 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,
}
std::cout << "\tscore: " << score << std::endl;
}
}
void CRNNRecognizer::LoadModel(const std::string &model_dir) {
void CRNNRecognizer::LoadModel(const std::string &model_dir) {
// AnalysisConfig config;
paddle_infer::Config config;
config.SetModel(model_dir + "/inference.pdmodel",
@ -125,10 +126,10 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,
config.DisableGlogInfo();
this->predictor_ = CreatePredictor(config);
}
}
cv::Mat CRNNRecognizer::GetRotateCropImage(
const cv::Mat &srcimage, std::vector<std::vector<int>> box) {
cv::Mat CRNNRecognizer::GetRotateCropImage(const cv::Mat &srcimage,
std::vector<std::vector<int>> box) {
cv::Mat image;
srcimage.copyTo(image);
std::vector<std::vector<int>> points = box;
@ -180,6 +181,6 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,
} else {
return dst_img;
}
}
}
} // namespace PaddleOCR