【导读】Keras作者François Chollet近期在Twitter上推荐了Github上的一个基于tf.keras的OCR项目Keras-OCR。该项目开箱即用,可直接执行OCR,也提供了端到端训练的流水线。
Keras作者François Chollet近期在Twitter上推荐了Github上的一个基于tf.keras的OCR项目Keras-OCR:
该项目地址为:
https://github.com/faustomorales/keras-ocr
目前该项目依赖环境为:
Python >= 3.6
TensorFlow >= 2.0.0
可以通过pip从Git或中央仓库进行安装:
# To install from master
pip install git+https://github.com/faustomorales/keras-ocr.git#egg=keras-ocr
# To install from PyPi
pip install keras-ocr
使用简单的几行就可以执行OCR:
import keras_ocr
# keras-ocr will automatically download pretrained
# weights for the detector and recognizer.
pipeline = keras_ocr.pipeline.Pipeline()
# Predictions is a list of (string, box) tuples.
predictions = pipeline.recognize(image='tests/test_image.jpg')
更多详细信息可阅读项目首页:
https://github.com/faustomorales/keras-ocr