How to use Tani04/segformer-b0-mars-testbed-4class with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="Tani04/segformer-b0-mars-testbed-4class")
# Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("Tani04/segformer-b0-mars-testbed-4class") model = SegformerForSemanticSegmentation.from_pretrained("Tani04/segformer-b0-mars-testbed-4class")
The community tab is the place to discuss and collaborate with the HF community!
