From 5935968f82018d68fff44946573d34cdf27db827 Mon Sep 17 00:00:00 2001 From: Lihe Yang Date: Thu, 25 Jan 2024 21:22:44 +0800 Subject: [PATCH] Update depth_anything.py --- metric_depth/zoedepth/models/base_models/depth_anything.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metric_depth/zoedepth/models/base_models/depth_anything.py b/metric_depth/zoedepth/models/base_models/depth_anything.py index 289f13b..e3e6d4a 100644 --- a/metric_depth/zoedepth/models/base_models/depth_anything.py +++ b/metric_depth/zoedepth/models/base_models/depth_anything.py @@ -338,7 +338,7 @@ class DepthAnythingCore(nn.Module): depth_anything = DPT_DINOv2(out_channels=[256, 512, 1024, 1024], use_clstoken=False) - state_dict = torch.load('../checkpoints/depth_anything_vitl14.pth', map_location='cpu') + state_dict = torch.load('./checkpoints/depth_anything_vitl14.pth', map_location='cpu') depth_anything.load_state_dict(state_dict) kwargs.update({'keep_aspect_ratio': force_keep_ar})