forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update image shape for angle class from [3,32,100] to [3,48,192]

This commit is contained in:
WenmuZhou 2020-10-27 16:20:17 +08:00
parent 3fb28d2a87
commit 5e1b7a184a
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deploy/android_demo/app/src/main/cpp/ocr_cls_process.cpp
View File

@ -20,7 +20,7 @@
#include <iostream>
#include <vector>
const std::vector<int> CLS_IMAGE_SHAPE = {3, 32, 100};
const std::vector<int> CLS_IMAGE_SHAPE = {3, 48, 192};
cv::Mat cls_resize_img(const cv::Mat &img) {
int imgC = CLS_IMAGE_SHAPE[0];

2
deploy/cpp_infer/include/preprocess_op.h
View File

@ -59,7 +59,7 @@ public:
class ClsResizeImg {
public:
virtual void Run(const cv::Mat &img, cv::Mat &resize_img,
const std::vector<int> &rec_image_shape = {3, 32, 320});
const std::vector<int> &rec_image_shape = {3, 48, 192});
};
} // namespace PaddleOCR

2
deploy/cpp_infer/src/ocr_cls.cpp
View File

@ -21,7 +21,7 @@ cv::Mat Classifier::Run(cv::Mat &img) {
img.copyTo(src_img);
cv::Mat resize_img;
std::vector<int> rec_image_shape = {3, 32, 100};
std::vector<int> rec_image_shape = {3, 48, 192};
int index = 0;
float wh_ratio = float(img.cols) / float(img.rows);

2
deploy/lite/cls_process.cc
View File

@ -17,7 +17,7 @@
#include <memory>
#include <string>
const std::vector<int> rec_image_shape{3, 32, 100};
const std::vector<int> rec_image_shape{3, 48, 192};
cv::Mat ClsResizeImg(cv::Mat img) {
int imgC, imgH, imgW;