- Notifications
You must be signed in to change notification settings - Fork506
A Unified Toolkit for Deep Learning Based Document Image Analysis
License
Layout-Parser/layout-parser
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
Layout Parser is deep learning based tool for document image layout analysis tasks.
Use pip or conda to install the library:
pip install layoutparser# Install Detectron2 for using DL Layout Detection Model# Please make sure the PyTorch version is compatible with# the installed Detectron2 version.pip install'git+https://github.com/facebookresearch/detectron2.git#egg=detectron2'# Install the ocr components when necessarypip install layoutparser[ocr]
This by default will install the CPU version of the Detectron2, and it should be able to run on most of the computers. But if you have a GPU, you can consider the GPU version of the Detectron2, referring to theofficial instructions.
We provide a series of examples for to help you start using the layout parser library:
Table OCR and Results Parsing:
layoutparser
can be used for conveniently OCR documents and convert the output in to structured data.Deep Layout Parsing Example: With the help of Deep Learning,
layoutparser
supports the analysis very complex documents and processing of the hierarchical structure in the layouts.
The images shown in the figure above are: a screenshot ofthis paper, an image from thePRIMA Layout Analysis Dataset, a screenshot of theWSJ website, and an image from theHJDataset.
With only 4 lines of code inlayoutparse
, you can unlock the information from complex documents that existing tools could not provide. You can either choose a deep learning model from theModelZoo, or load the model that you trained on your own. And use the following code to predict the layout as well as visualize it:
>>>importlayoutparseraslp>>>model=lp.Detectron2LayoutModel('lp://PrimaLayout/mask_rcnn_R_50_FPN_3x/config')>>>layout=model.detect(image)# You need to load the image somewhere else, e.g., image = cv2.imread(...)>>>lp.draw_box(image,layout,)# With extra configurations
If you findlayoutparser
helpful to your work, please consider citing our tool using the following BibTeX entry.
@misc{shen2020layoutparser, author = {Zejiang Shen and Ruochen Zhang and Melissa Dell}, title = {LayoutParser}, howpublished = {\url{https://github.com/Layout-Parser/layout-parser}}, year = {2020}}
About
A Unified Toolkit for Deep Learning Based Document Image Analysis
Topics
Resources
License
Code of conduct
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Packages0
Uh oh!
There was an error while loading.Please reload this page.