smp-hub
/

dpt-large-ade20k

Image Segmentation

segmentation-models-pytorch

model_hub_mixin

pytorch_model_hub_mixin

semantic-segmentation

Model card Files Files and versions

qubvel-hf commited on Apr 6

Commit

c79f56e

·

verified ·

1 Parent(s): 1d636b2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ from PIL import Image
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load pretrained model and preprocessing function
-checkpoint = "smp-hub/smp-hub/dpt-large-ade20k"
 model = smp.from_pretrained(checkpoint).eval().to(device)
 preprocessing = A.Compose.from_pretrained(checkpoint)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load pretrained model and preprocessing function
+checkpoint = "smp-hub/dpt-large-ade20k"
 model = smp.from_pretrained(checkpoint).eval().to(device)
 preprocessing = A.Compose.from_pretrained(checkpoint)