opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-10-24 02:06:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support fixed length TRT prediction

Browse Source
This commit is contained in:
root
2021-01-08 09:34:29 +00:00
parent 1f99a63524
commit 24f8dfb7e1
6 changed files with 47 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deploy/cpp_infer/src/ocr_cls.cpp
+1 -1
View File
@@ -25,7 +25,7 @@ cv::Mat Classifier::Run(cv::Mat &img) {
int index = 0;
float wh_ratio = float(img.cols) / float(img.rows);
this->resize_op_.Run(img, resize_img, cls_image_shape);
this->resize_op_.Run(img, resize_img, this->use_tensorrt_, cls_image_shape);
this->normalize_op_.Run(&resize_img, this->mean_, this->scale_,
this->is_scale_);