未验证 提交 fbf0fab5 编写于 作者: D dyning 提交者: GitHub

Merge pull request #346 from LDOUBLEV/fixocr

Support to recognize space text
......@@ -235,9 +235,9 @@ RunDetModel(std::shared_ptr<PaddlePredictor> predictor, cv::Mat img,
}
cv::Mat cbuf_map(shape_out[2], shape_out[3], CV_8UC1,
reinterpret_cast<unsigned char *> cbuf);
reinterpret_cast<unsigned char *>(cbuf));
cv::Mat pred_map(shape_out[2], shape_out[3], CV_32F,
reinterpret_cast<float *> pred);
reinterpret_cast<float *>(pred));
const double threshold = double(Config["det_db_thresh"]) * 255;
const double maxvalue = 255;
......@@ -335,6 +335,7 @@ int main(int argc, char **argv) {
auto rec_predictor = loadModel(rec_model_file);
auto charactor_dict = ReadDict(dict_path);
charactor_dict.push_back(" ");
cv::Mat srcimg = cv::imread(img_path, cv::IMREAD_COLOR);
auto boxes = RunDetModel(det_predictor, srcimg, Config);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册