Update README.md
Browse files
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
|
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
|