linhaotong commited on
Commit ·
9380192
1
Parent(s): 4d55109
update model card
Browse files
README.md
CHANGED
|
@@ -34,7 +34,7 @@ prompt_depth_path = "assets/example_images/arkit_depth.png"
|
|
| 34 |
image = load_image(image_path).to(DEVICE)
|
| 35 |
prompt_depth = load_depth(prompt_depth_path).to(DEVICE) # 192x256, ARKit LiDAR depth in meters
|
| 36 |
|
| 37 |
-
model = PromptDA.from_pretrained("depth-anything/prompt-depth-anything-
|
| 38 |
depth = model.predict(image, prompt_depth) # HxW, depth in meters
|
| 39 |
|
| 40 |
save_depth(depth, prompt_depth=prompt_depth, image=image)
|
|
|
|
| 34 |
image = load_image(image_path).to(DEVICE)
|
| 35 |
prompt_depth = load_depth(prompt_depth_path).to(DEVICE) # 192x256, ARKit LiDAR depth in meters
|
| 36 |
|
| 37 |
+
model = PromptDA.from_pretrained("depth-anything/prompt-depth-anything-vits-transparent").to(DEVICE).eval()
|
| 38 |
depth = model.predict(image, prompt_depth) # HxW, depth in meters
|
| 39 |
|
| 40 |
save_depth(depth, prompt_depth=prompt_depth, image=image)
|