diff --git a/deploy/cpp_infer/include/ocr_rec.h b/deploy/cpp_infer/include/ocr_rec.h index d271b9b303..ff80ba5299 100644 --- a/deploy/cpp_infer/include/ocr_rec.h +++ b/deploy/cpp_infer/include/ocr_rec.h @@ -89,7 +89,7 @@ private: // pre-process CrnnResizeImg resize_op_; Normalize normalize_op_; - Permute_batch permute_op_; + PermuteBatch permute_op_; // post-process PostProcessor post_processor_; diff --git a/deploy/cpp_infer/include/preprocess_op.h b/deploy/cpp_infer/include/preprocess_op.h index c93473300b..31217de301 100644 --- a/deploy/cpp_infer/include/preprocess_op.h +++ b/deploy/cpp_infer/include/preprocess_op.h @@ -44,7 +44,7 @@ public: virtual void Run(const cv::Mat *im, float *data); }; -class Permute_batch { +class PermuteBatch { public: virtual void Run(const std::vector imgs, float *data); }; diff --git a/deploy/cpp_infer/src/preprocess_op.cpp b/deploy/cpp_infer/src/preprocess_op.cpp index 4c2c2f7a54..14e8bd1d84 100644 --- a/deploy/cpp_infer/src/preprocess_op.cpp +++ b/deploy/cpp_infer/src/preprocess_op.cpp @@ -40,7 +40,7 @@ void Permute::Run(const cv::Mat *im, float *data) { } } -void Permute_batch::Run(const std::vector imgs, float *data) { +void PermuteBatch::Run(const std::vector imgs, float *data) { for (int j = 0; j < imgs.size(); j ++){ int rh = imgs[j].rows; int rw = imgs[j].cols;