rename inference model name

4 years ago · fa0ad0f4fd
parent 13cc1f3c41
commit fa0ad0f4fd
3 changed files with 3 additions and 3 deletions
--- a/deploy/cpp_infer/src/ocr_cls.cpp
+++ b/deploy/cpp_infer/src/ocr_cls.cpp
@ -81,7 +81,7 @@ cv::Mat Classifier::Run(cv::Mat &img) {

 void Classifier::LoadModel(const std::string &model_dir) {
  AnalysisConfig config;
-  config.SetModel(model_dir + "/model", model_dir + "/params");
+  config.SetModel(model_dir + "/cls.pdmodel", model_dir + "/cls.pdiparams");

  if (this->use_gpu_) {
    config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
--- a/deploy/cpp_infer/src/ocr_det.cpp
+++ b/deploy/cpp_infer/src/ocr_det.cpp
@ -18,7 +18,7 @@ namespace PaddleOCR {

 void DBDetector::LoadModel(const std::string &model_dir) {
  AnalysisConfig config;
-  config.SetModel(model_dir + "/model", model_dir + "/params");
+  config.SetModel(model_dir + "/det.pdmodel", model_dir + "/det.pdiparams");

  if (this->use_gpu_) {
    config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
--- a/deploy/cpp_infer/src/ocr_rec.cpp
+++ b/deploy/cpp_infer/src/ocr_rec.cpp
@ -103,7 +103,7 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,

 void CRNNRecognizer::LoadModel(const std::string &model_dir) {
  AnalysisConfig config;
-  config.SetModel(model_dir + "/model", model_dir + "/params");
+  config.SetModel(model_dir + "/rec.pdmodel", model_dir + "/rec.pdiparams");

  if (this->use_gpu_) {
    config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);