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