mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00

Files

Harry Stern c119600d6e Move image tensor to device in trocr example (#2063 )

Signed-off-by: Harry Stern <harry@harrystern.net>

2024-04-15 06:50:32 +02:00

assets

docs: add trocr examples (#1692 )

2024-02-10 16:14:50 +01:00

image_processor.rs

Add support for TrOCR Model (#1303 )

2023-11-09 18:49:17 +01:00

main.rs

Move image tensor to device in trocr example (#2063 )

2024-04-15 06:50:32 +02:00

readme.md

docs: add trocr examples (#1692 )

2024-02-10 16:14:50 +01:00

readme.md

candle-trocr

TrOCR is a transformer OCR Model. In this example it is used to transcribe image text. See the associated model card for details on the model itself.

Supported models include:

--which base: small handwritten OCR model.
--which large: large handwritten OCR model.
--which base-printed: small printed OCR model.
--which large-printed: large printed OCR model.

Running an example

cargo run --example trocr --release -- --image candle-examples/examples/trocr/assets/trocr.png
cargo run --example trocr --release -- --which large --image candle-examples/examples/trocr/assets/trocr.png
cargo run --example trocr --release -- --which base-printed --image candle-examples/examples/trocr/assets/noto.png
cargo run --example trocr --release -- --which large-printed --image candle-examples/examples/trocr/assets/noto.png

Outputs

industry , Mr. Brown commented icily . " Let us have a
industry , " Mr. Brown commented icily . " Let us have a
THE QUICK BROWN FOR JUMPS OVER THE LAY DOG
THE QUICK BROWN FOX JUMPS OVER THE LAZY DOG