Mercurial > hgrepos > Python2 > PyMuPDF
diff mupdf-source/thirdparty/tesseract/src/training/unicharset/lstmtester.cpp @ 2:b50eed0cc0ef upstream
ADD: MuPDF v1.26.7: the MuPDF source as downloaded by a default build of PyMuPDF 1.26.4.
The directory name has changed: no version number in the expanded directory now.
| author | Franz Glasner <fzglas.hg@dom66.de> |
|---|---|
| date | Mon, 15 Sep 2025 11:43:07 +0200 |
| parents | |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mupdf-source/thirdparty/tesseract/src/training/unicharset/lstmtester.cpp Mon Sep 15 11:43:07 2025 +0200 @@ -0,0 +1,157 @@ +/////////////////////////////////////////////////////////////////////// +// File: lstmtester.cpp +// Description: Top-level line evaluation class for LSTM-based networks. +// Author: Ray Smith +// +// (C) Copyright 2016, Google Inc. +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// http://www.apache.org/licenses/LICENSE-2.0 +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +/////////////////////////////////////////////////////////////////////// + +#include "lstmtester.h" +#include <iomanip> // for std::setprecision +#include <thread> // for std::thread +#include "fileio.h" // for LoadFileLinesToStrings + +namespace tesseract { + +LSTMTester::LSTMTester(int64_t max_memory) : test_data_(max_memory) {} + +// Loads a set of lstmf files that were created using the lstm.train config to +// tesseract into memory ready for testing. Returns false if nothing was +// loaded. The arg is a filename of a file that lists the filenames. +bool LSTMTester::LoadAllEvalData(const char *filenames_file) { + std::vector<std::string> filenames; + if (!LoadFileLinesToStrings(filenames_file, &filenames)) { + tprintf("Failed to load list of eval filenames from %s\n", filenames_file); + return false; + } + return LoadAllEvalData(filenames); +} + +// Loads a set of lstmf files that were created using the lstm.train config to +// tesseract into memory ready for testing. Returns false if nothing was +// loaded. +bool LSTMTester::LoadAllEvalData(const std::vector<std::string> &filenames) { + test_data_.Clear(); + bool result = test_data_.LoadDocuments(filenames, CS_SEQUENTIAL, nullptr); + total_pages_ = test_data_.TotalPages(); + return result; +} + +// Runs an evaluation asynchronously on the stored data and returns a string +// describing the results of the previous test. +std::string LSTMTester::RunEvalAsync(int iteration, const double *training_errors, + const TessdataManager &model_mgr, int training_stage) { + std::string result; + if (total_pages_ == 0) { + result += "No test data at iteration " + std::to_string(iteration); + return result; + } + if (!LockIfNotRunning()) { + result += "Previous test incomplete, skipping test at iteration " + std::to_string(iteration); + return result; + } + // Save the args. + std::string prev_result = test_result_; + test_result_ = ""; + if (training_errors != nullptr) { + test_iteration_ = iteration; + test_training_errors_ = training_errors; + test_model_mgr_ = model_mgr; + test_training_stage_ = training_stage; + std::thread t(&LSTMTester::ThreadFunc, this); + t.detach(); + } else { + UnlockRunning(); + } + return prev_result; +} + +// Runs an evaluation synchronously on the stored data and returns a string +// describing the results. +std::string LSTMTester::RunEvalSync(int iteration, const double *training_errors, + const TessdataManager &model_mgr, int training_stage, + int verbosity) { + LSTMTrainer trainer; + trainer.InitCharSet(model_mgr); + TFile fp; + if (!model_mgr.GetComponent(TESSDATA_LSTM, &fp) || !trainer.DeSerialize(&model_mgr, &fp)) { + return "Deserialize failed"; + } + int eval_iteration = 0; + double char_error = 0.0; + double word_error = 0.0; + int error_count = 0; + while (error_count < total_pages_) { + const ImageData *trainingdata = test_data_.GetPageBySerial(eval_iteration); + trainer.SetIteration(++eval_iteration); + NetworkIO fwd_outputs, targets; + Trainability result = trainer.PrepareForBackward(trainingdata, &fwd_outputs, &targets); + if (result != UNENCODABLE) { + char_error += trainer.NewSingleError(tesseract::ET_CHAR_ERROR); + word_error += trainer.NewSingleError(tesseract::ET_WORD_RECERR); + ++error_count; + if (verbosity > 1 || (verbosity > 0 && result != PERFECT)) { + tprintf("Truth:%s\n", trainingdata->transcription().c_str()); + std::vector<int> ocr_labels; + std::vector<int> xcoords; + trainer.LabelsFromOutputs(fwd_outputs, &ocr_labels, &xcoords); + std::string ocr_text = trainer.DecodeLabels(ocr_labels); + tprintf("OCR :%s\n", ocr_text.c_str()); + if (verbosity > 2 || (verbosity > 1 && result != PERFECT)) { + tprintf("Line BCER=%f, BWER=%f\n\n", + trainer.NewSingleError(tesseract::ET_CHAR_ERROR), + trainer.NewSingleError(tesseract::ET_WORD_RECERR)); + } + } + } + } + char_error *= 100.0 / total_pages_; + word_error *= 100.0 / total_pages_; + std::stringstream result; + result.imbue(std::locale::classic()); + result << std::fixed << std::setprecision(3); + if (iteration != 0 || training_stage != 0) { + result << "At iteration " << iteration + << ", stage " << training_stage << ", "; + } + result << "BCER eval=" << char_error << ", BWER eval=" << word_error; + return result.str(); +} + +// Helper thread function for RunEvalAsync. +// LockIfNotRunning must have returned true before calling ThreadFunc, and +// it will call UnlockRunning to release the lock after RunEvalSync completes. +void LSTMTester::ThreadFunc() { + test_result_ = + RunEvalSync(test_iteration_, test_training_errors_, test_model_mgr_, test_training_stage_, + /*verbosity*/ 0); + UnlockRunning(); +} + +// Returns true if there is currently nothing running, and takes the lock +// if there is nothing running. +bool LSTMTester::LockIfNotRunning() { + std::lock_guard<std::mutex> lock(running_mutex_); + if (async_running_) { + return false; + } + async_running_ = true; + return true; +} + +// Releases the running lock. +void LSTMTester::UnlockRunning() { + std::lock_guard<std::mutex> lock(running_mutex_); + async_running_ = false; +} + +} // namespace tesseract
