Hugging Face's logo

cmarkea
/
dit-base-layout-detection

Image Segmentation
Transformers
Safetensors
beit
Inference Endpoints
Model card Files Community
2
dit-base-layout-detection / README.md
Cyrile's picture
Cyrile
Update README.md
136732b verified
preview code
|
raw
history blame
1.62 kB
metadata
library_name: transformers
tags: []

Model Card for Model ID

Model Details

Model Description

This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.

  • Developed by: [More Information Needed]
  • Funded by [optional]: [More Information Needed]
  • Shared by [optional]: [More Information Needed]
  • Model type: [More Information Needed]
  • Language(s) (NLP): [More Information Needed]
  • License: [More Information Needed]
  • Finetuned from model [optional]: [More Information Needed]

Model Sources [optional]

  • Repository: [More Information Needed]
  • Paper [optional]: [More Information Needed]
  • Demo [optional]: [More Information Needed]

Uses

Direct Use 

import torch
from transformers import AutoImageProcessor, AutoModel

img_proc = AutoImageProcessor.from_pretrained(
    "ArkeaIAF/dit-base-layout-detection"
)
model = AutoModel.from_pretrained(
    "ArkeaIAF/dit-base-layout-detection"
)

with torch.inference_mode():
    input_ids = img_proc(img, return_tensors='pt')
    segmentation = model(**input_ids)

segmentation_mask = img_proc.post_process_semantic_segmentation(
    segmentation,
    target_sizes=[img.size[::-1]]
)