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

This commit is contained in:
zhoujun 2020-10-27 20:17:55 +08:00 committed by GitHub
parent 6947196684
commit 3573388b7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -59,7 +59,7 @@ public:
class ClsResizeImg { class ClsResizeImg {
public: public:
virtual void Run(const cv::Mat &img, cv::Mat &resize_img, 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 } // namespace PaddleOCR

View File

@ -21,7 +21,7 @@ cv::Mat Classifier::Run(cv::Mat &img) {
img.copyTo(src_img); img.copyTo(src_img);
cv::Mat resize_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; int index = 0;
float wh_ratio = float(img.cols) / float(img.rows); float wh_ratio = float(img.cols) / float(img.rows);