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
Fix size
Browse files- preprocessor_config.json +1 -1
preprocessor_config.json
CHANGED
|
@@ -14,5 +14,5 @@
|
|
| 14 |
],
|
| 15 |
"reduce_labels": false,
|
| 16 |
"resample": 2,
|
| 17 |
-
"size":
|
| 18 |
}
|
|
|
|
| 14 |
],
|
| 15 |
"reduce_labels": false,
|
| 16 |
"resample": 2,
|
| 17 |
+
"size": 1024
|
| 18 |
}
|