From ed10e2d2895ca59a45c58b441583afb85cf17e6b Mon Sep 17 00:00:00 2001 From: masterAllen Date: Sat, 7 Sep 2024 12:05:41 +0800 Subject: [PATCH] Fix OCRBeamSearchDecoder bug: a risk of assigning value to NULL. --- modules/text/src/ocr_beamsearch_decoder.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/modules/text/src/ocr_beamsearch_decoder.cpp b/modules/text/src/ocr_beamsearch_decoder.cpp index 2bf630da71..c0a36980f3 100644 --- a/modules/text/src/ocr_beamsearch_decoder.cpp +++ b/modules/text/src/ocr_beamsearch_decoder.cpp @@ -338,9 +338,12 @@ class OCRBeamSearchDecoderImpl CV_FINAL : public OCRBeamSearchDecoder double lp = score_segmentation( beam[0].segmentation, out_sequence ); // fill other (dummy) output parameters - component_rects->push_back(Rect(0,0,src.cols,src.rows)); - component_texts->push_back(out_sequence); - component_confidences->push_back((float)exp(lp)); + if (component_rects != NULL) + component_rects->push_back(Rect(0,0,src.cols,src.rows)); + if (component_texts != NULL) + component_texts->push_back(out_sequence); + if (component_confidences != NULL) + component_confidences->push_back((float)exp(lp)); return; }