修复cpp_infer的bug

deploy/cpp_infer/include/ocr_cls.h

@@ -45,6 +45,7 @@ public:
     this->cpu_math_library_num_threads_ = cpu_math_library_num_threads;
     this->use_mkldnn_ = use_mkldnn;
     this->use_zero_copy_run_ = use_zero_copy_run;
+
     this->cls_thresh = cls_thresh;
 
     LoadModel(model_dir);
@@ -63,7 +64,7 @@ private:
   int gpu_mem_ = 4000;
   int cpu_math_library_num_threads_ = 4;
   bool use_mkldnn_ = false;
-
+  bool use_zero_copy_run_ = false;
   double cls_thresh = 0.5;
 
   std::vector<float> mean_ = {0.5f, 0.5f, 0.5f};

deploy/cpp_infer/src/ocr_cls.cpp

@@ -96,7 +96,7 @@ void Classifier::LoadModel(const std::string &model_dir) {
   }
 
   // false for zero copy tensor
-  config.SwitchUseFeedFetchOps(false);
+  config.SwitchUseFeedFetchOps(!this->use_zero_copy_run_);
   // true for multiple input
   config.SwitchSpecifyInputNames(true);
 

deploy/cpp_infer/tools/config.txt

@@ -13,6 +13,10 @@ det_db_box_thresh  0.5
 det_db_unclip_ratio  2.0
 det_model_dir  ./inference/det_db
 
+# cls config
+cls_model_dir ./inference/cls
+cls_thresh  0.9
+
 # rec config
 rec_model_dir  ./inference/rec_crnn
 char_list_file ../../ppocr/utils/ppocr_keys_v1.txt