yupengchengg147 commited on
Commit
3435ee5
·
verified ·
1 Parent(s): b2af22b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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/St4rTrack")
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(), "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
  )
 
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
  )