Modelpublic

anyformat/doclayout-yolo-docstructbench

sourceHugging Faceapache-2.0updated 8mo agoView on Hugging Face

0likes1kdownloads

Model Card

DocLayout-YOLO - Docstructbench

Document layout detection model. Paper: DocLayout-YOLO

Classes (10)

—title
—plain_text
—abandon
—figure
—figure_caption
—table
—table_caption
—table_footnote
—isolate_formula
—formula_caption

Usage

python

pip install anyformat-doclayout

python

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}")