Intel
/

dpt-beit-base-384

Depth Estimation

Model card Files Files and versions

nielsr HF Staff commited on Nov 28, 2023

Commit

37e4dfa

·

1 Parent(s): 6614d7d

Update README.md (#5)

- Update README.md (a6875881d41f3442904fc8372537878cc2023386)

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")