split cpp inference

deploy/cpp_infer/include/config.h

@@ -1,123 +0,0 @@
-// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-#pragma once
-
-#include <iomanip>
-#include <iostream>
-#include <map>
-#include <ostream>
-#include <string>
-#include <vector>
-
-#include "include/utility.h"
-
-namespace PaddleOCR {
-
-class OCRConfig {
-public:
-  explicit OCRConfig(const std::string &config_file) {
-    config_map_ = LoadConfig(config_file);
-
-    this->use_gpu = bool(stoi(config_map_["use_gpu"]));
-
-    this->gpu_id = stoi(config_map_["gpu_id"]);
-
-    this->gpu_mem = stoi(config_map_["gpu_mem"]);
-
-    this->cpu_math_library_num_threads =
-        stoi(config_map_["cpu_math_library_num_threads"]);
-
-    this->use_mkldnn = bool(stoi(config_map_["use_mkldnn"]));
-
-    this->max_side_len = stoi(config_map_["max_side_len"]);
-
-    this->det_db_thresh = stod(config_map_["det_db_thresh"]);
-
-    this->det_db_box_thresh = stod(config_map_["det_db_box_thresh"]);
-
-    this->det_db_unclip_ratio = stod(config_map_["det_db_unclip_ratio"]);
-
-    this->use_polygon_score = bool(stoi(config_map_["use_polygon_score"]));
-
-    this->det_model_dir.assign(config_map_["det_model_dir"]);
-
-    this->rec_model_dir.assign(config_map_["rec_model_dir"]);
-
-    this->char_list_file.assign(config_map_["char_list_file"]);
-
-    this->use_angle_cls = bool(stoi(config_map_["use_angle_cls"]));
-
-    this->cls_model_dir.assign(config_map_["cls_model_dir"]);
-
-    this->cls_thresh = stod(config_map_["cls_thresh"]);
-
-    this->visualize = bool(stoi(config_map_["visualize"]));
-
-    this->use_tensorrt = bool(stoi(config_map_["use_tensorrt"]));
-
-    this->use_fp16 = bool(stod(config_map_["use_fp16"]));
-  }
-
-  bool use_gpu = false;
-
-  int gpu_id = 0;
-
-  int gpu_mem = 4000;
-
-  int cpu_math_library_num_threads = 1;
-
-  bool use_mkldnn = false;
-
-  int max_side_len = 960;
-
-  double det_db_thresh = 0.3;
-
-  double det_db_box_thresh = 0.5;
-
-  double det_db_unclip_ratio = 2.0;
-
-  bool use_polygon_score = false;
-
-  std::string det_model_dir;
-
-  std::string rec_model_dir;
-
-  bool use_angle_cls;
-
-  std::string char_list_file;
-
-  std::string cls_model_dir;
-
-  double cls_thresh;
-
-  bool visualize = true;
-
-  bool use_tensorrt = false;
-
-  bool use_fp16 = false;
-
-  void PrintConfigInfo();
-
-private:
-  // Load configuration
-  std::map<std::string, std::string> LoadConfig(const std::string &config_file);
-
-  std::vector<std::string> split(const std::string &str,
-                                 const std::string &delim);
-
-  std::map<std::string, std::string> config_map_;
-};
-
-} // namespace PaddleOCR

deploy/cpp_infer/include/ocr_det.h

@@ -38,15 +38,20 @@ namespace PaddleOCR {
 
 class DBDetector {
 public:
-  explicit DBDetector(const std::string &model_dir, const bool &use_gpu,
-                      const int &gpu_id, const int &gpu_mem,
-                      const int &cpu_math_library_num_threads,
-                      const bool &use_mkldnn, const int &max_side_len,
-                      const double &det_db_thresh,
-                      const double &det_db_box_thresh,
-                      const double &det_db_unclip_ratio,
-                      const bool &use_polygon_score, const bool &visualize,
-                      const bool &use_tensorrt, const bool &use_fp16) {
+  explicit DBDetector(const std::string &model_dir,
+                      const bool &use_gpu=false,
+                      const int &gpu_id=0,
+                      const int &gpu_mem=4000,
+                      const int &cpu_math_library_num_threads=1,
+                      const bool &use_mkldnn=false,
+                      const int &max_side_len=960,
+                      const double &det_db_thresh=0.3,
+                      const double &det_db_box_thresh=0.5,
+                      const double &det_db_unclip_ratio=2.0,
+                      const bool &use_polygon_score=false,
+                      const bool &visualize=false,
+                      const bool &use_tensorrt=false, 
+                      const bool &use_fp16=false) {
     this->use_gpu_ = use_gpu;
     this->gpu_id_ = gpu_id;
     this->gpu_mem_ = gpu_mem;

deploy/cpp_infer/include/ocr_rec.h

@@ -62,8 +62,7 @@ public:
   // Load Paddle inference model
   void LoadModel(const std::string &model_dir);
 
-  void Run(std::vector<std::vector<std::vector<int>>> boxes, cv::Mat &img,
-           Classifier *cls);
+  void Run(cv::Mat &img);
 
 private:
   std::shared_ptr<Predictor> predictor_;