Instructions to use vikp/layout_segmenter with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use vikp/layout_segmenter with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="vikp/layout_segmenter")# Load model directly from transformers import AutoProcessor, AutoModelForTokenClassification processor = AutoProcessor.from_pretrained("vikp/layout_segmenter") model = AutoModelForTokenClassification.from_pretrained("vikp/layout_segmenter") - Inference
- Notebooks
- Google Colab
- Kaggle
dataset being fine-tuned on?
#2
by guruprasadr - opened
I presume you've fine-tuned the layoutlmv3 model (provided by microsoft) on the classification task - to annotate every block within the document.
What dataset has been used to fine-tune the base layoutlmv3 model?
guruprasadr changed discussion status to closed
I presume you've fine-tuned on the DocLayNet dataset (https://huggingface.co/datasets/ds4sd/DocLayNet-v1.1?row=0) - on the 11 different classes.