Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,6 @@ import torch
|
|
| 35 |
|
| 36 |
from depth_anything_v2.dpt import DepthAnythingV2
|
| 37 |
|
| 38 |
-
# take depth-anything-v2-large as an example
|
| 39 |
model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
|
| 40 |
model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location='cpu'))
|
| 41 |
model.eval()
|
|
|
|
| 35 |
|
| 36 |
from depth_anything_v2.dpt import DepthAnythingV2
|
| 37 |
|
|
|
|
| 38 |
model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
|
| 39 |
model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location='cpu'))
|
| 40 |
model.eval()
|