From b931343d35c9e9768a8f20af4687c4d59cd1b089 Mon Sep 17 00:00:00 2001 From: MistEO Date: Sun, 30 Jan 2022 01:29:18 +0800 Subject: [PATCH] =?UTF-8?q?perf.=E6=9B=B4=E6=8D=A2=E9=A3=9E=E6=A1=A8OCR=20?= =?UTF-8?q?det=E6=A8=A1=E5=9E=8B=EF=BC=8C=E6=8F=90=E9=AB=98=E5=B9=B2?= =?UTF-8?q?=E5=91=98=E5=90=8D=E8=AF=86=E5=88=AB=E7=8E=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 3rdparty/resource/PaddleOCR/det/version.txt | 2 +- src/MeoAssistant/OcrPack.cpp | 9 ++++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/3rdparty/resource/PaddleOCR/det/version.txt b/3rdparty/resource/PaddleOCR/det/version.txt index 3e35e11c98..02c07004e9 100644 --- a/3rdparty/resource/PaddleOCR/det/version.txt +++ b/3rdparty/resource/PaddleOCR/det/version.txt @@ -1 +1 @@ -ch_ppocr_mobile_v2.0_det_prune_infer \ No newline at end of file +ch_PP-OCRv2_det_slim_quant_infer diff --git a/src/MeoAssistant/OcrPack.cpp b/src/MeoAssistant/OcrPack.cpp index 74adeaa821..e8d5136340 100644 --- a/src/MeoAssistant/OcrPack.cpp +++ b/src/MeoAssistant/OcrPack.cpp @@ -61,6 +61,11 @@ std::vector asst::OcrPack::recognize(const cv::Mat image, const std::vector result; std::string log_str_raw; std::string log_str_proc; + +#ifdef ASST_DEBUG + cv::Mat draw = image.clone(); +#endif + for (size_t i = 0; i != size; ++i) { // the box rect like ↓ // 0 - 1 @@ -80,7 +85,9 @@ std::vector asst::OcrPack::recognize(const cv::Mat image, const float score = *(scores + i); TextRect tr{ score, rect, text }; - +#ifdef ASST_DEBUG + cv::rectangle(draw, utils::make_rect(rect), cv::Scalar(0, 0, 255), 2); +#endif log_str_raw += tr.to_string() + ", "; if (!pred || pred(tr)) { log_str_proc += tr.to_string() + ", ";