depth-anything
/

prompt-depth-anything-vits-transparent

Depth Estimation

Model card Files Files and versions Community

linhaotong commited on 4 days ago

Commit

9380192

·

1 Parent(s): 4d55109

update model card

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -34,7 +34,7 @@ prompt_depth_path = "assets/example_images/arkit_depth.png"
 image = load_image(image_path).to(DEVICE)
 prompt_depth = load_depth(prompt_depth_path).to(DEVICE) # 192x256, ARKit LiDAR depth in meters
-model = PromptDA.from_pretrained("depth-anything/prompt-depth-anything-vitl").to(DEVICE).eval()
 depth = model.predict(image, prompt_depth) # HxW, depth in meters
 save_depth(depth, prompt_depth=prompt_depth, image=image)

 image = load_image(image_path).to(DEVICE)
 prompt_depth = load_depth(prompt_depth_path).to(DEVICE) # 192x256, ARKit LiDAR depth in meters
+model = PromptDA.from_pretrained("depth-anything/prompt-depth-anything-vits-transparent").to(DEVICE).eval()
 depth = model.predict(image, prompt_depth) # HxW, depth in meters
 save_depth(depth, prompt_depth=prompt_depth, image=image)