WeiChow
/

dptv2_l_vit

Model card Files Files and versions Community

WeiChow commited on 2 days ago

Commit

b3d0349

•

1 Parent(s): 5245f4c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ model_configs = {
     'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
 }
-encoder = 'vitb' # or 'vits', 'vitb'
 model = DepthAnythingV2(**model_configs[encoder])
 model.load_state_dict(torch.load(f'depth_anything_v2_{encoder}.pth?download=true', map_location='cpu'))

     'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
 }
+encoder = 'vitl' # or 'vits', 'vitb'
 model = DepthAnythingV2(**model_configs[encoder])
 model.load_state_dict(torch.load(f'depth_anything_v2_{encoder}.pth?download=true', map_location='cpu'))