b2d0d5cb1c | 1 month ago | |
---|---|---|
.circleci | 1 month ago | |
.github/workflows | 1 month ago | |
repo | ||
sbb_binarize | 2 years ago | |
.gitignore | ||
.gitkeep | ||
.gitmodules | ||
CHANGELOG.md | 2 years ago | |
Dockerfile | 1 month ago | |
LICENSE | ||
Makefile | 1 month ago | |
README.md | 1 month ago | |
make.sh | ||
ocrd-tool.json | ||
pyproject.toml | 1 month ago | |
requirements.txt | 1 month ago |
README.md
sbb_binarization
Document Image Binarization
Installation
Python 3.8-3.11
with Tensorflow <2.13
are currently supported. While newer versions might also work, we currently don't test this.
You can either install from PyPI via
pip install sbb-binarization
or clone the repository, enter it and install (editable) with
git clone git@github.com:qurator-spk/sbb_binarization.git
cd sbb_binarization; pip install -e .
Alternatively, download the prebuilt image from Dockerhub:
docker pull ocrd/sbb_binarization
Models
Pre-trained models can be downloaded from the locations below. We also provide models and model cards on 🤗
Version | Format | Download |
---|---|---|
2021-03-09 | SavedModel |
https://github.com/qurator-spk/sbb_binarization/releases/download/v0.0.11/saved_model_2021_03_09.zip |
2021-03-09 | HDF5 |
https://qurator-data.de/sbb_binarization/2021-03-09/models.tar.gz |
2020-01-16 | SavedModel |
https://github.com/qurator-spk/sbb_binarization/releases/download/v0.0.11/saved_model_2020_01_16.zip |
2020-01-16 | HDF5 |
https://qurator-data.de/sbb_binarization/2020-01-16/models.tar.gz |
With OCR-D, you can also use the Resource Manager, e.g.
ocrd resmgr download ocrd-sbb-binarize "*"
Usage
sbb_binarize \
-m <path to directory containing model files> \
<input image> \
<output image>
Note: the output image MUST use either .tif
or .png
as file extension to produce a binary image. Input images can also be JPEG.
Images containing a lot of border noise (black pixels) should be cropped beforehand to improve the quality of results.
Example
sbb_binarize -m /path/to/model/ myimage.tif myimage-bin.tif
To use the OCR-D interface:
ocrd-sbb-binarize -I INPUT_FILE_GRP -O OCR-D-IMG-BIN -P model default
Testing
For simple smoke tests, the following will
-
download models
-
download test data
-
run the OCR-D wrapper (on page and region level):
make models make test
How to cite
If you find this tool useful in your work, please consider citing our paper:
@inproceedings{hip23rezanezhad2,
author = {Vahid Rezanezhad and Konstantin Baierer and Clemens Neudecker},
editor = {Apostolos Antonacopoulos and Christian Clausner and Maud Ehrmann and Kai Labusch and Clemens Neudecker},
title = {A hybrid CNN-Transformer Model for Historical Document Image Binarization},
booktitle = {Proceedings of the 7th International Workshop on Historical Document Imaging and Processing {HIP} 2023,
San José, CA, USA, August 26, 2023},
year = {2023},
url = {https://doi.org/10.1145/3604951.3605508}
}