Views
No views yet
1from unidepth.models import UniDepthV2
2import torch
3
4model = UniDepthV2.from_pretrained("lpiccinelli/unidepth-v2-vitl14")
5ckpt = torch.load("unidepth-v2-vitl14-rsrd-pothole.pth", map_location="cpu")
6model.load_state_dict(ckpt["model"], strict=False)