Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

nvidia
/
segformer-b5-finetuned-cityscapes-1024-1024

Image Segmentation
Transformers
PyTorch
google-tensorflow TensorFlow
segformer
vision
Model card Files Files and versions
xet
Community
5

Instructions to use nvidia/segformer-b5-finetuned-cityscapes-1024-1024 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use nvidia/segformer-b5-finetuned-cityscapes-1024-1024 with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-segmentation", model="nvidia/segformer-b5-finetuned-cityscapes-1024-1024")
    # Load model directly
    from transformers import AutoImageProcessor, SegformerForSemanticSegmentation
    
    processor = AutoImageProcessor.from_pretrained("nvidia/segformer-b5-finetuned-cityscapes-1024-1024")
    model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b5-finetuned-cityscapes-1024-1024")
  • Inference
  • Notebooks
  • Google Colab
  • Kaggle
New discussion
Resources
  • PR & discussions documentation
  • Code of Conduct
  • Hub documentation

How to generate config.json and model.bin

#5 opened over 1 year ago by
zzq8888666

Update to feature_extractor_type and reduce_labels due to deprecation notices.

#4 opened over 2 years ago by
theHolger

Adding `safetensors` variant of this model

#3 opened about 3 years ago by
SFconvertbot
Company
TOS Privacy About Careers
Website
Models Datasets Spaces Pricing Docs