apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-16 05:30:58 +08:00
Code Issues Actions 6 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

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

@@ -20,10 +20,14 @@ namespace vision {
cv::Mat VisOcr(const cv::Mat& im, const OCRResult& ocr_result,
const float score_threshold) {
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++) {
if (ocr_result.rec_scores[n] < score_threshold) {
continue;
if (have_score) {
if (ocr_result.rec_scores[n] < score_threshold) {
continue;
}
}
cv::Point rook_points[4];