11 stable releases

new 1.1.10 Mar 25, 2024
1.1.9 Oct 11, 2023
1.1.7 Jul 26, 2023
1.1.5 Jun 20, 2023
1.0.0 Feb 24, 2022

#27 in Machine learning

Download history 25/week @ 2023-12-06 33/week @ 2023-12-13 35/week @ 2023-12-20 5/week @ 2023-12-27 54/week @ 2024-01-03 16/week @ 2024-01-10 41/week @ 2024-01-17 36/week @ 2024-01-24 43/week @ 2024-01-31 55/week @ 2024-02-07 41/week @ 2024-02-14 87/week @ 2024-02-21 123/week @ 2024-02-28 266/week @ 2024-03-06 92/week @ 2024-03-13 91/week @ 2024-03-20

582 downloads per month

MIT license

51KB
731 lines

rusty-tesseract

A Rust wrapper for Google Tesseract

Crates.io Libraries.io dependency status for GitHub repo GitHub last commit GitHub Stars Forks Issues

Installation

Add the following line to your Cargo.toml file:

rusty-tesseract = "1.1.10"

Description

  • Brings all relevant command-line tesseract functionality to Rust
  • Partly based on the Python wrapper for tesseract (i.e. https://github.com/madmaze/pytesseract)
  • Enables testing a pre-trained tesseract model and outputting the results in different formats such as strings, bounding boxes, dicts, or dataframes.

Dependencies

Tesseract: https://github.com/tesseract-ocr/tesseract

Usage

1. Read Image

Create an Image object by specifying a path or alternatively a DynamicImage from the image crate https://docs.rs/image/latest/image/

// you can use the from_path function
let _ = Image::from_path("img/string.png");


// or instantiate Image from a DynamicImage
let dynamic_image = ImageReader::open("img/string.png")
    .unwrap()
    .decode()
    .unwrap();
let img = Image::from_dynamic_image(&dynamic_image).unwrap();

2. Set tesseract parameters

Set tesseract parameters using the Args struct.

let default_args = Args::default();

// the default parameters are
/*
Args {
    lang: "eng",
    dpi: Some(150),
    psm: Some(3),
    oem: Some(3),
}
*/

// fill your own argument struct if needed
// Optional arguments are ignored if set to `None`
let mut my_args = Args {
    //model language (tesseract default = 'eng')
    //available languages can be found by running 'rusty_tesseract::get_tesseract_langs()'
    lang: "eng",

    //map of config variables
    //this example shows a whitelist for the normal alphabet. Multiple arguments are allowed.
    //available arguments can be found by running 'rusty_tesseract::get_tesseract_config_parameters()'
    config_variables: HashMap::from([(
            "tessedit_char_whitelist".into(),
            "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ".into(),
        )]),
    dpi: Some(150),       // specify DPI for input image
    psm: Some(6),         // define page segmentation mode 6 (i.e. "Assume a single uniform block of text")
    oem: Some(3),         // define optical character recognition mode 3 (i.e. "Default, based on what is available")
};

3. Get the tesseract model output

Choose either string, bounding box or data output:

// define parameters
let mut my_args = Args {
    lang: "eng",
    config_variables: HashMap::from([(
            "tessedit_char_whitelist".into(),
            "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ".into(),
        )]),
    dpi: Some(150),
    psm: Some(6),
    oem: Some(3)
};

// string output
let output = rusty_tesseract::image_to_string(&img, &my_args).unwrap();
    println!("The String output is: {:?}", output);



// image_to_boxes creates a BoxOutput containing the parsed output from Tesseract when using the "makebox" Parameter
let box_output = rusty_tesseract::image_to_boxes(&img, &my_args).unwrap();
println!(
    "The first boxfile symbol is: {}",
    box_output.boxes[0].symbol
);
println!("The full boxfile output is:\n{}", box_output.output);

// image_to_data creates a DataOutput containing the parsed output from Tesseract when using the "TSV" Parameter
let data_output = rusty_tesseract::image_to_data(&img, &my_args).unwrap();
let first_text_line = &data_output.data[4];
println!(
    "The first text is '{}' with confidence {}",
    first_text_line.text, first_text_line.conf
);
println!("The full data output is:\n{}", data_output.output);

Get information about tesseract

//tesseract version
let tesseract_version = rusty_tesseract::get_tesseract_version().unwrap();
println!("The tesseract version is: {:?}", tesseract_version);

//available languages
let tesseract_langs = rusty_tesseract::get_tesseract_langs().unwrap();
println!("The available languages are: {:?}", tesseract_langs);

//available config parameters
let parameters = rusty_tesseract::get_tesseract_config_parameters().unwrap();
println!("Example config parameter: {}", parameters.config_parameters.first().unwrap());

Contributing

  1. Fork the repository
  2. Create a new feature branch (git checkout -b my-feature-branch-name)
  3. Commit your new changes (git commit -m 'commit message' <changed-file>)
  4. Push changes to the branch (git push origin my-feature-branch-name)
  5. Create a Pull Request

Dependencies

~13–23MB
~186K SLoC