提交 5e1b7a18 编写于 作者: W WenmuZhou

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

上级 3fb28d2a
...@@ -20,7 +20,7 @@ ...@@ -20,7 +20,7 @@
#include <iostream> #include <iostream>
#include <vector> #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) { cv::Mat cls_resize_img(const cv::Mat &img) {
int imgC = CLS_IMAGE_SHAPE[0]; int imgC = CLS_IMAGE_SHAPE[0];
......
...@@ -59,7 +59,7 @@ public: ...@@ -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
\ No newline at end of file
...@@ -21,7 +21,7 @@ cv::Mat Classifier::Run(cv::Mat &img) { ...@@ -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);
......
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,7 @@
#include <memory> #include <memory>
#include <string> #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) { cv::Mat ClsResizeImg(cv::Mat img) {
int imgC, imgH, imgW; int imgC, imgH, imgW;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册