解决tessert-ocr自动文本方向识别时内存泄漏的问题

This commit is contained in:
luoliangyi 2023-07-12 17:30:46 +08:00
parent a4e92b8d59
commit 97e5ff8c17
5 changed files with 58 additions and 54 deletions

View File

@ -97,30 +97,34 @@ static Pix* RemoveEnclosingCircle(Pix* pixs) {
* On return the blocks list owns all the constructed page layout. * On return the blocks list owns all the constructed page layout.
*/ */
int Tesseract::SegmentPage(const STRING* input_file, BLOCK_LIST* blocks, int Tesseract::SegmentPage(const STRING* input_file, BLOCK_LIST* blocks,
Tesseract* osd_tess, OSResults* osr) { Tesseract* osd_tess, OSResults* osr)
{
ASSERT_HOST(pix_binary_ != nullptr); ASSERT_HOST(pix_binary_ != nullptr);
int width = pixGetWidth(pix_binary_); int width = pixGetWidth(pix_binary_);
int height = pixGetHeight(pix_binary_); int height = pixGetHeight(pix_binary_);
// Get page segmentation mode. // Get page segmentation mode.
auto pageseg_mode = static_cast<PageSegMode>( auto pageseg_mode = static_cast<PageSegMode>(static_cast<int>(tessedit_pageseg_mode));
static_cast<int>(tessedit_pageseg_mode));
// If a UNLV zone file can be found, use that instead of segmentation. // If a UNLV zone file can be found, use that instead of segmentation.
if (!PSM_COL_FIND_ENABLED(pageseg_mode) && if (!PSM_COL_FIND_ENABLED(pageseg_mode) && input_file != nullptr && input_file->length() > 0)
input_file != nullptr && input_file->length() > 0) { {
STRING name = *input_file; STRING name = *input_file;
const char* lastdot = strrchr(name.string(), '.'); const char* lastdot = strrchr(name.string(), '.');
if (lastdot != nullptr) if (lastdot != nullptr)
name[lastdot - name.string()] = '\0'; name[lastdot - name.string()] = '\0';
read_unlv_file(name, width, height, blocks); read_unlv_file(name, width, height, blocks);
} }
if (blocks->empty()) {
if (blocks->empty())
{
// No UNLV file present. Work according to the PageSegMode. // No UNLV file present. Work according to the PageSegMode.
// First make a single block covering the whole image. // First make a single block covering the whole image.
BLOCK_IT block_it(blocks); BLOCK_IT block_it(blocks);
auto* block = new BLOCK("", true, 0, 0, 0, 0, width, height); auto* block = new BLOCK("", true, 0, 0, 0, 0, width, height);
block->set_right_to_left(right_to_left()); block->set_right_to_left(right_to_left());
block_it.add_to_end(block); block_it.add_to_end(block);
} else { }
else
{
// UNLV file present. Use PSM_SINGLE_BLOCK. // UNLV file present. Use PSM_SINGLE_BLOCK.
pageseg_mode = PSM_SINGLE_BLOCK; pageseg_mode = PSM_SINGLE_BLOCK;
} }
@ -133,15 +137,13 @@ int Tesseract::SegmentPage(const STRING* input_file, BLOCK_LIST* blocks,
BLOBNBOX_LIST diacritic_blobs; BLOBNBOX_LIST diacritic_blobs;
int auto_page_seg_ret_val = 0; int auto_page_seg_ret_val = 0;
TO_BLOCK_LIST to_blocks; TO_BLOCK_LIST to_blocks;
if (PSM_OSD_ENABLED(pageseg_mode) || PSM_BLOCK_FIND_ENABLED(pageseg_mode) || if (PSM_OSD_ENABLED(pageseg_mode) || PSM_BLOCK_FIND_ENABLED(pageseg_mode) || PSM_SPARSE(pageseg_mode))
PSM_SPARSE(pageseg_mode)) { {
auto_page_seg_ret_val = AutoPageSeg( auto_page_seg_ret_val = AutoPageSeg(pageseg_mode, blocks, &to_blocks, enable_noise_removal ? &diacritic_blobs : nullptr, osd_tess, osr);
pageseg_mode, blocks, &to_blocks,
enable_noise_removal ? &diacritic_blobs : nullptr, osd_tess, osr);
if (pageseg_mode == PSM_OSD_ONLY) if (pageseg_mode == PSM_OSD_ONLY)
return auto_page_seg_ret_val; return auto_page_seg_ret_val;
// To create blobs from the image region bounds uncomment this line: // To create blobs from the image region bounds uncomment this line:
// to_blocks.clear(); // Uncomment to go back to the old mode. to_blocks.clear(); // Uncomment to go back to the old mode.
} else { } else {
deskew_ = FCOORD(1.0f, 0.0f); deskew_ = FCOORD(1.0f, 0.0f);
reskew_ = FCOORD(1.0f, 0.0f); reskew_ = FCOORD(1.0f, 0.0f);
@ -213,6 +215,8 @@ int Tesseract::AutoPageSeg(PageSegMode pageseg_mode, BLOCK_LIST* blocks,
#if 1 #if 1
pixDestroy(&photomask_pix); pixDestroy(&photomask_pix);
pixDestroy(&musicmask_pix); pixDestroy(&musicmask_pix);
delete finder;
blocks->clear();
return 0; return 0;
#else #else
int result = 0; int result = 0;