Support fixed length TRT prediction

2026-09-24 23:33:08 +08:00 · 2021-01-08 09:34:29 +00:00
parent 1f99a63524
commit 24f8dfb7e1
6 changed files with 47 additions and 19 deletions
@@ -47,18 +47,20 @@ public:
 class ResizeImgType0 {
 public:
  virtual void Run(const cv::Mat &img, cv::Mat &resize_img, int max_size_len,
-                   float &ratio_h, float &ratio_w);
+                   float &ratio_h, float &ratio_w, bool use_tensorrt);
 };

 class CrnnResizeImg {
 public:
  virtual void Run(const cv::Mat &img, cv::Mat &resize_img, float wh_ratio,
+                   bool use_tensorrt = false,
                   const std::vector<int> &rec_image_shape = {3, 32, 320});
 };

 class ClsResizeImg {
 public:
  virtual void Run(const cv::Mat &img, cv::Mat &resize_img,
+                   bool use_tensorrt = false,
                   const std::vector<int> &rec_image_shape = {3, 48, 192});
 };