How to use google/deeplabv3_mobilenet_v2_1.0_513 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="google/deeplabv3_mobilenet_v2_1.0_513")
# Load model directly from transformers import AutoImageProcessor, MobileNetV2ForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("google/deeplabv3_mobilenet_v2_1.0_513") model = MobileNetV2ForSemanticSegmentation.from_pretrained("google/deeplabv3_mobilenet_v2_1.0_513", device_map="auto")
SFconvertbot