vimassaru/teethsegmentation
Viewer • Updated • 113 • 47 • 1
How to use vimassaru/segformer-b2-finetuned-teeth-segmentation with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-segmentation", model="vimassaru/segformer-b2-finetuned-teeth-segmentation") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("vimassaru/segformer-b2-finetuned-teeth-segmentation", dtype="auto")