LiheYoung commited on
Commit
e96fd0c
·
verified ·
1 Parent(s): c5507f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -1
README.md CHANGED
@@ -4,6 +4,18 @@ license: apache-2.0
4
 
5
  # Depth Anything V2 for Metric Depth Estimation
6
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  ## Usage
8
 
9
  ### Prepraration
@@ -14,7 +26,7 @@ cd Depth-Anything-V2/metric_depth
14
  pip install -r requirements.txt
15
  ```
16
 
17
- Download the checkpoint and put them under the `checkpoints` directory.
18
 
19
  ### Use our models
20
  ```python
 
4
 
5
  # Depth Anything V2 for Metric Depth Estimation
6
 
7
+ # Pre-trained Models
8
+
9
+ We provide **six metric depth models** of three scales for indoor and outdoor scenes, respectively.
10
+
11
+ | Base Model | Params | Indoor (Hypersim) | Outdoor (Virtual KITTI 2) |
12
+ |:-|-:|:-:|:-:|
13
+ | Depth-Anything-V2-Small | 24.8M | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-Hypersim-Small/resolve/main/depth_anything_v2_metric_hypersim_vits.pth?download=true) | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-VKITTI-Small/resolve/main/depth_anything_v2_metric_vkitti_vits.pth?download=true) |
14
+ | Depth-Anything-V2-Base | 97.5M | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-Hypersim-Base/resolve/main/depth_anything_v2_metric_hypersim_vitb.pth?download=true) | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-VKITTI-Base/resolve/main/depth_anything_v2_metric_vkitti_vitb.pth?download=true) |
15
+ | Depth-Anything-V2-Large | 335.3M | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-Hypersim-Large/resolve/main/depth_anything_v2_metric_hypersim_vitl.pth?download=true) | [Download](https://huggingface.co/depth-anything/Depth-Anything-V2-Metric-VKITTI-Large/resolve/main/depth_anything_v2_metric_vkitti_vitl.pth?download=true) |
16
+
17
+ *We recommend to first try our larger models (if computational cost is affordable) and the indoor version.*
18
+
19
  ## Usage
20
 
21
  ### Prepraration
 
26
  pip install -r requirements.txt
27
  ```
28
 
29
+ Download the checkpoints listed [here](#pre-trained-models) and put them under the `checkpoints` directory.
30
 
31
  ### Use our models
32
  ```python