Skip to content

Commit

Permalink
Update Ocrs project link in README and reformat paragraphs
Browse files Browse the repository at this point in the history
  • Loading branch information
robertknight committed Jan 24, 2024
1 parent 2d16768 commit 2c696aa
Showing 1 changed file with 11 additions and 7 deletions.
18 changes: 11 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,16 +1,20 @@
# ocrs-models

This project contains PyTorch models for the
[**ocrs**](https://github.com/robertknight/rten/tree/main/ocrs) OCR engine.
This project contains tools for training PyTorch models for use with the
[**Ocrs**](https://github.com/robertknight/ocrs/) OCR engine.

The ocrs engine splits text detection and recognition into three phases, each
of which corresponds to a different model in this repository:

1. **Text detection**: This is a semantic segmentation model which classifies
each pixel in a greyscale input image as text/non-text. Consumers then post-process clusters of text pixels to get oriented bounding boxes for words.
2. **Layout analysis (VERY WIP)**: This is a graph model which takes word bounding boxes as input nodes and
classifies each node's relation to nearby nodes (eg. start / middle / end of line)
3. **Text recognition**: This is a CRNN model that takes a greyscale image of a text line as input and returns a sequence of characters.
1. **Text detection**: This is a semantic segmentation model which classifies
each pixel in a greyscale input image as text/non-text. Consumers then
post-process clusters of text pixels to get oriented bounding boxes for
words.
2. **Layout analysis (VERY WIP)**: This is a graph model which takes word
bounding boxes as input nodes and classifies each node's relation to nearby
nodes (eg. start / middle / end of line)
3. **Text recognition**: This is a CRNN model that takes a greyscale image of a
text line as input and returns a sequence of characters.

All models can be exported to ONNX for downstream use.

Expand Down

0 comments on commit 2c696aa

Please sign in to comment.