You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
eynollah/README.md

122 lines
5.9 KiB
Markdown

# Eynollah
7 months ago
> Document Layout Analysis with Deep Learning and Heuristics
2 years ago
[![PyPI Version](https://img.shields.io/pypi/v/eynollah)](https://pypi.org/project/eynollah/)
2 years ago
[![GH Actions Test](https://github.com/qurator-spk/eynollah/actions/workflows/test-eynollah.yml/badge.svg)](https://github.com/qurator-spk/eynollah/actions/workflows/test-eynollah.yml)
2 years ago
[![License: ASL](https://img.shields.io/github/license/qurator-spk/eynollah)](https://opensource.org/license/apache-2-0/)
7 months ago
[![DOI](https://img.shields.io/badge/DOI-10.1145%2F3604951.3605513-red)](https://doi.org/10.1145/3604951.3605513)
4 years ago
![](https://user-images.githubusercontent.com/952378/102350683-8a74db80-3fa5-11eb-8c7e-f743f7d6eae2.jpg)
2 years ago
## Features
* Support for up to 10 segmentation classes:
* background, [page border](https://ocr-d.de/en/gt-guidelines/trans/lyRand.html), [text region](https://ocr-d.de/en/gt-guidelines/trans/lytextregion.html#textregionen__textregion_), [text line](https://ocr-d.de/en/gt-guidelines/pagexml/pagecontent_xsd_Complex_Type_pc_TextLineType.html), [header](https://ocr-d.de/en/gt-guidelines/trans/lyUeberschrift.html), [image](https://ocr-d.de/en/gt-guidelines/trans/lyBildbereiche.html), [separator](https://ocr-d.de/en/gt-guidelines/trans/lySeparatoren.html), [marginalia](https://ocr-d.de/en/gt-guidelines/trans/lyMarginalie.html), [initial](https://ocr-d.de/en/gt-guidelines/trans/lyInitiale.html), [table](https://ocr-d.de/en/gt-guidelines/trans/lyTabellen.html)
2 years ago
* Support for various image optimization operations:
* cropping (border detection), binarization, deskewing, dewarping, scaling, enhancing, resizing
2 years ago
* Text line segmentation to bounding boxes or polygons (contours) including for curved lines and vertical text
7 months ago
* Detection of reading order (left-to-right or right-to-left)
2 years ago
* Output in [PAGE-XML](https://github.com/PRImA-Research-Lab/PAGE-XML)
2 years ago
* [OCR-D](https://github.com/qurator-spk/eynollah#use-as-ocr-d-processor) interface
2 years ago
4 months ago
:warning: Eynollah development is currently focused on achieving the best possible quality of results for a wide variety of historical documents and therefore processing can be very slow. We aim to improve this, but contributions are always welcome.
7 months ago
3 years ago
## Installation
7 months ago
Python `3.8-3.11` with Tensorflow `2.12-2.15` on Linux are currently supported.
3 years ago
For (limited) GPU support the CUDA toolkit needs to be installed.
3 years ago
7 months ago
You can either install from PyPI
3 years ago
2 years ago
```
pip install eynollah
```
3 years ago
2 years ago
or clone the repository, enter it and install (editable) with
3 years ago
2 years ago
```
git clone git@github.com:qurator-spk/eynollah.git
cd eynollah; pip install -e .
```
2 years ago
2 years ago
Alternatively, you can run `make install` or `make install-dev` for editable installation.
2 years ago
2 years ago
## Models
Pre-trained models can be downloaded from [qurator-data.de](https://qurator-data.de/eynollah/) or [huggingface](https://huggingface.co/SBB?search_models=eynollah).
2 years ago
7 months ago
## Train
🚧 **Work in progress**
In case you want to train your own model, have a look at [`sbb_pixelwise_segmentation`](https://github.com/qurator-spk/sbb_pixelwise_segmentation).
2 years ago
3 years ago
## Usage
The command-line interface can be called like this:
```sh
2 years ago
eynollah \
7 months ago
-i <single image file> | -di <directory containing image files> \
2 years ago
-o <output directory> \
7 months ago
-m <directory containing model files> \
2 years ago
[OPTIONS]
2 years ago
```
2 years ago
The following options can be used to further configure the processing:
3 years ago
2 years ago
| option | description |
|----------|:-------------|
| `-fl` | full layout analysis including all steps and segmentation classes |
| `-light` | lighter and faster but simpler method for main region detection and deskewing |
2 years ago
| `-tab` | apply table detection |
| `-ae` | apply enhancement (the resulting image is saved to the output directory) |
2 years ago
| `-as` | apply scaling |
1 year ago
| `-cl` | apply contour detection for curved text lines instead of bounding boxes |
2 years ago
| `-ib` | apply binarization (the resulting image is saved to the output directory) |
| `-ep` | enable plotting (MUST always be used with `-sl`, `-sd`, `-sa`, `-si` or `-ae`) |
| `-ho` | ignore headers for reading order dectection |
| `-si <directory>` | save image regions detected to this directory |
2 years ago
| `-sd <directory>` | save deskewed image to this directory |
| `-sl <directory>` | save layout prediction as plot to this directory |
2 years ago
| `-sp <directory>` | save cropped page image to this directory |
| `-sa <directory>` | save all (plot, enhanced/binary image, layout) to this directory |
3 years ago
4 months ago
If no option is set, the tool performs layout detection of main regions (background, text, images, separators and marginals).
Best quality output is produced when RGB images are used as input rather than greyscale or binarized images.
3 years ago
#### Use as OCR-D processor
7 months ago
🚧 **Work in progress**
4 years ago
2 years ago
Eynollah ships with a CLI interface to be used as [OCR-D](https://ocr-d.de) processor.
4 years ago
2 years ago
In this case, the source image file group with (preferably) RGB images should be used as input like this:
4 years ago
2 years ago
```
ocrd-eynollah-segment -I OCR-D-IMG -O SEG-LINE -P models
```
3 years ago
2 years ago
Any image referenced by `@imageFilename` in PAGE-XML is passed on directly to Eynollah as a processor, so that e.g.
4 years ago
2 years ago
```
ocrd-eynollah-segment -I OCR-D-IMG-BIN -O SEG-LINE -P models
```
3 years ago
2 years ago
uses the original (RGB) image despite any binarization that may have occured in previous OCR-D processing steps
7 months ago
#### Additional documentation
Please check the [wiki](https://github.com/qurator-spk/eynollah/wiki).
## How to cite
If you find this tool useful in your work, please consider citing our paper:
```bibtex
7 months ago
@inproceedings{hip23rezanezhad,
title = {Document Layout Analysis with Deep Learning and Heuristics},
author = {Rezanezhad, Vahid and Baierer, Konstantin and Gerber, Mike and Labusch, Kai and Neudecker, Clemens},
booktitle = {Proceedings of the 7th International Workshop on Historical Document Imaging and Processing {HIP} 2023,
San José, CA, USA, August 25-26, 2023},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
year = {2023},
pages = {73--78},
url = {https://doi.org/10.1145/3604951.3605513}
}
```