docs: correct MASt3R metric checkpoint references
#1
by unit27research - opened
README.md
CHANGED
|
@@ -39,7 +39,7 @@ Gihub page: https://github.com/naver/mast3r/
|
|
| 39 |
|
| 40 |
| Modelname | Training resolutions | Head | Encoder | Decoder |
|
| 41 |
|-------------|----------------------|------|---------|---------|
|
| 42 |
-
|
|
| 43 |
|
| 44 |
# How to use
|
| 45 |
|
|
@@ -50,7 +50,7 @@ To load the model:
|
|
| 50 |
from mast3r.model import AsymmetricMASt3R
|
| 51 |
import torch
|
| 52 |
|
| 53 |
-
model = AsymmetricMASt3R.from_pretrained("naver/
|
| 54 |
|
| 55 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 56 |
model.to(device)
|
|
|
|
| 39 |
|
| 40 |
| Modelname | Training resolutions | Head | Encoder | Decoder |
|
| 41 |
|-------------|----------------------|------|---------|---------|
|
| 42 |
+
| MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric | 512x384, 512x336, 512x288, 512x256, 512x160 | CatMLP+DPT | ViT-L | ViT-B |
|
| 43 |
|
| 44 |
# How to use
|
| 45 |
|
|
|
|
| 50 |
from mast3r.model import AsymmetricMASt3R
|
| 51 |
import torch
|
| 52 |
|
| 53 |
+
model = AsymmetricMASt3R.from_pretrained("naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric")
|
| 54 |
|
| 55 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 56 |
model.to(device)
|