DocLayout-YOLO: Enhancing Document Layout Analysis through Diverse Synthetic Data and Global-to-Local Adaptive Perception
Paper • 2410.12628 • Published • 41
DocLayout-YOLO - Docstructbench
Document layout detection model. Paper: DocLayout-YOLO
Classes (10)
titleplain_textabandonfigurefigure_captiontabletable_captiontable_footnoteisolate_formulaformula_caption
Usage
pip install anyformat-doclayout
from anyformat.doclayout import DocLayoutModel, download_converted
# Download weights from this repo
weights = download_converted("docstructbench")
# Run inference
model = DocLayoutModel(weights)
results = model.predict("document.png")
for det in results:
print(f"{det['class_name']}: {det['confidence']:.2f}")
- Downloads last month
- 234
# Couldn't find a valid YOLO version tag. # Replace XX with the correct version. from ultralytics import YOLOvXX model = YOLOvXX.from_pretrained("anyformat/doclayout-yolo-docstructbench") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)