apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-17 14:11:14 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Bug Fix] fixed ocr visualize error (#2090)

Browse Source
This commit is contained in:
DefTruth
2023-07-07 17:43:08 +08:00
committed by GitHub
parent ad1f46f7d9
commit 4c1e80b723
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/vision/visualize/ocr.cc
View File

@@ -20,11 +20,15 @@ namespace vision {
cv::Mat VisOcr(const cv::Mat& im, const OCRResult& ocr_result, cv::Mat VisOcr(const cv::Mat& im, const OCRResult& ocr_result,
const float score_threshold) { const float score_threshold) {
auto vis_im = im.clone(); auto vis_im = im.clone();
bool have_score =
(ocr_result.boxes.size() == ocr_result.rec_scores.size());
for (int n = 0; n < ocr_result.boxes.size(); n++) { for (int n = 0; n < ocr_result.boxes.size(); n++) {
if (have_score) {
if (ocr_result.rec_scores[n] < score_threshold) { if (ocr_result.rec_scores[n] < score_threshold) {
continue; continue;
} }
}
cv::Point rook_points[4]; cv::Point rook_points[4];
for (int m = 0; m < 4; m++) { for (int m = 0; m < 4; m++) {