Model Details
Full Model IDnvidia/segformer-b5-finetuned-cityscapes-1024-1024
Pipeline / Taskimage-segmentation
Librarytransformers
Downloads (all-time)37.2K
Likes43
Last Modified8/9/2022
Author / Orgnvidia
PrivateNo — public
⚡ Quick Usage (Python)
Using the 🤗 Transformers library. Install with pip install transformers
from transformers import pipeline
# Load the model
pipe = pipeline("image-segmentation", model="nvidia/segformer-b5-finetuned-cityscapes-1024-1024")
# Run inference
result = pipe("Your input here")
print(result)🏷️ Tags
transformerspytorchtfsegformervisionimage-segmentationdataset:cityscapesarxiv:2105.15203license:otherendpoints_compatibledeploy:azureregion:us
More image-segmentation Models
See all →🚀 Use This Model
Access model files, inference API, and full documentation on Hugging Face.
Open on Hugging Face →Browse Model Files ↗← Browse All Models🤖 Task: image-segmentation
This model is designed for the image-segmentation task. Explore more models for this use case.
All image-segmentation Models →📊 Popularity
⬇ Downloads37.2K
❤️ Community Likes43
🛠️ Requirements
- →Install: pip install transformers
- →Python 3.8+ recommended for Transformers.
- →GPU (CUDA) speeds up inference significantly.
- →Use model.half() for fp16 on limited VRAM.