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
6.5 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
3 months ago
:warning: 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 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
| 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 |
| `-tab` | apply table detection |
| `-ae` | apply enhancement (the resulting image is saved to the output directory) |
| `-as` | apply scaling |
| `-cl` | apply contour detection for curved text lines instead of bounding boxes |
| `-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 |
| `-sd <directory>` | save deskewed image to this directory |
| `-sl <directory>` | save layout prediction as plot to this directory |
| `-sp <directory>` | save cropped page image to this directory |
| `-sa <directory>` | save all (plot, enhanced/binary image, layout) to this directory |
3 years ago
3 months ago
If no option is set, the tool performs layout detection of main regions (background, text, images, separators and marginals).
3 months ago
The best output quality 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}
}
```