Update README.md
Browse files
README.md
CHANGED
@@ -33,7 +33,7 @@ First install [St4rTrack](https://github.com/HavenFeng/St4RTrack). To load the m
|
|
33 |
from dust3r.models import AsymmetricCroCo3DStereo
|
34 |
|
35 |
# Loads default Seq checkpoint
|
36 |
-
model = AsymmetricCroCo3DStereo.from_pretrained("yupengchengg147/
|
37 |
```
|
38 |
|
39 |
Run the following code to load the checkpoint trained with Pair Mode:
|
@@ -45,12 +45,12 @@ import os
|
|
45 |
from dust3r.model import AsymmetricCroCo3DStereo
|
46 |
|
47 |
# Create a temporary directory for the model files
|
48 |
-
temp_dir = os.path.join(tempfile.gettempdir(), "
|
49 |
os.makedirs(temp_dir, exist_ok=True)
|
50 |
|
51 |
# Download the config and model files from the Pair subfolder
|
52 |
config_path = hf_hub_download(
|
53 |
-
repo_id="yupengchengg147/
|
54 |
filename="Pair/config.json",
|
55 |
cache_dir=temp_dir
|
56 |
)
|
|
|
33 |
from dust3r.models import AsymmetricCroCo3DStereo
|
34 |
|
35 |
# Loads default Seq checkpoint
|
36 |
+
model = AsymmetricCroCo3DStereo.from_pretrained("yupengchengg147/St4RTrack")
|
37 |
```
|
38 |
|
39 |
Run the following code to load the checkpoint trained with Pair Mode:
|
|
|
45 |
from dust3r.model import AsymmetricCroCo3DStereo
|
46 |
|
47 |
# Create a temporary directory for the model files
|
48 |
+
temp_dir = os.path.join(tempfile.gettempdir(), "St4RTrack_pair")
|
49 |
os.makedirs(temp_dir, exist_ok=True)
|
50 |
|
51 |
# Download the config and model files from the Pair subfolder
|
52 |
config_path = hf_hub_download(
|
53 |
+
repo_id="yupengchengg147/St4RTrack",
|
54 |
filename="Pair/config.json",
|
55 |
cache_dir=temp_dir
|
56 |
)
|