Intel
/

dpt-beit-base-384

Depth Estimation

Model card Files Files and versions

Update README.md

#5

by nielsr HF Staff - opened Nov 28, 2023

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -28,8 +28,8 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-processor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
 # prepare image for the model
 inputs = processor(images=image, return_tensors="pt")

 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
+processor = DPTImageProcessor.from_pretrained("Intel/dpt-beit-base-384")
+model = DPTForDepthEstimation.from_pretrained("Intel/dpt-beit-base-384")
 # prepare image for the model
 inputs = processor(images=image, return_tensors="pt")