Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,6 @@ Pre-trained checkpoints for **MASt3R** (Matching And Stereo 3D Reconstruction) w
|
|
| 29 |
| `dune_vit_small_448` | 448x448 | ViT-S/14 | ~1.3 GB | Fast inference |
|
| 30 |
| `dune_vit_base_336` | 336x336 | ViT-B/14 | ~1.7 GB | Balanced speed/quality |
|
| 31 |
| `dune_vit_base_448` | 448x448 | ViT-B/14 | ~1.7 GB | High quality |
|
| 32 |
-
| `mast3r_vit_large` | 512x512 | ViT-L/14 | ~2.9 GB | Maximum precision |
|
| 33 |
|
| 34 |
## Architecture
|
| 35 |
|
|
@@ -60,13 +59,13 @@ from huggingface_hub import hf_hub_download
|
|
| 60 |
|
| 61 |
# Download encoder
|
| 62 |
encoder = hf_hub_download(
|
| 63 |
-
repo_id="
|
| 64 |
filename="dune_vit_small_336/encoder.safetensors"
|
| 65 |
)
|
| 66 |
|
| 67 |
# Download decoder
|
| 68 |
decoder = hf_hub_download(
|
| 69 |
-
repo_id="
|
| 70 |
filename="dune_vit_small_336/decoder.safetensors"
|
| 71 |
)
|
| 72 |
```
|
|
|
|
| 29 |
| `dune_vit_small_448` | 448x448 | ViT-S/14 | ~1.3 GB | Fast inference |
|
| 30 |
| `dune_vit_base_336` | 336x336 | ViT-B/14 | ~1.7 GB | Balanced speed/quality |
|
| 31 |
| `dune_vit_base_448` | 448x448 | ViT-B/14 | ~1.7 GB | High quality |
|
|
|
|
| 32 |
|
| 33 |
## Architecture
|
| 34 |
|
|
|
|
| 59 |
|
| 60 |
# Download encoder
|
| 61 |
encoder = hf_hub_download(
|
| 62 |
+
repo_id="Aedelon/dunemast3r-models-fp16",
|
| 63 |
filename="dune_vit_small_336/encoder.safetensors"
|
| 64 |
)
|
| 65 |
|
| 66 |
# Download decoder
|
| 67 |
decoder = hf_hub_download(
|
| 68 |
+
repo_id="Aedelon/dunemast3r-models-fp16",
|
| 69 |
filename="dune_vit_small_336/decoder.safetensors"
|
| 70 |
)
|
| 71 |
```
|