Views
No views yet
DeepLabV3SemanticSegment, ResNet-101, VOC 21 classes) converted from torchvision.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from PIL import Image
5from kerasformers.models.deeplabv3 import DeepLabV3SemanticSegment, DeepLabV3ImageProcessor
6
7model = DeepLabV3SemanticSegment.from_weights("kerasformers/deeplabv3_resnet101_coco_voc")
8processor = DeepLabV3ImageProcessor.from_weights("kerasformers/deeplabv3_resnet101_coco_voc")
9
10image = Image.open("your_image.jpg").convert("RGB")
11output = model(processor(image)["pixel_values"], training=False)
12result = processor.post_process_semantic_segmentation(
13 output, target_size=(image.height, image.width)
14)
15print(result["segmentation"].shape)from_weights("kerasformers/<variant>"):| Variant | Hub | Backbone |
|---|---|---|
deeplabv3_resnet50_coco_voc | kerasformers/deeplabv3_resnet50_coco_voc | ResNet-50 |
deeplabv3_resnet101_coco_voc | kerasformers/deeplabv3_resnet101_coco_voc | ResNet-101 |
KERAS_BACKEND before importing Keras / kerasformers.