Spaces:
Sleeping
Sleeping
from transformers import DPTFeatureExtractor, DPTForSemanticSegmentation | |
from PIL import Image | |
import requests | |
url = "http://images.cocodataset.org/val2017/000000039769.jpg" | |
image = Image.open(requests.get(url, stream=True).raw) | |
feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large-ade") | |
model = DPTForSemanticSegmentation.from_pretrained("Intel/dpt-large-ade") | |
inputs = feature_extractor(images=image, return_tensors="pt") | |
outputs = model(**inputs) | |
logits = outputs.logits |