Nicholai/SpaTrackerV2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix_requirements

Browse Source
This commit is contained in:
xiaoyuxi 2025-07-08 16:32:40 +08:00
parent be230bbe99
commit fd1a134647
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -4,7 +4,7 @@
[Yuxi Xiao](https://henry123-boy.github.io/), [Jianyuan Wang](https://jytime.github.io/), [Nan Xue](https://xuenan.net/), [Nikita Karaev](https://nikitakaraevv.github.io/), [Iurii Makarov](https://linkedin.com/in/lvoursl), [Bingyi Kang](https://bingykang.github.io/), [Xin Zhu](https://openreview.net/profile?id=~Xing_Zhu2), [Hujun Bao](http://www.cad.zju.edu.cn/home/bao/), [Yujun Shen](https://shenyujun.github.io/), [Xiaowei Zhou](https://www.xzhou.me/)
### [Project Page]() | [BibTeX]() | [Goolge Drive]()
### [Project Page](https://spatialtracker.github.io/) | [BibTeX]() | [Goolge Drive](https://drive.google.com/drive/u/1/folders/1GYeC639gA23N_OiytGHXTUCSYrbM0pOo?dmr=1&ec=wgc-drive-globalnav-goto)
<!-- [Paper V1]() | [Paper V2]() | [Paper V3]() | -->

2
inference.py
View File

@ -42,7 +42,7 @@ if __name__ == "__main__":
fps = int(args.fps)
mask_dir = args.data_dir + f"/{args.video_name}.png"
os.environ["VGGT_DIR"] = "/mnt/bn/xyxdata/home/codes/my_projs/SpaTrack2/checkpoints/MagicCam/vggt_release_7.7/exp/ckpt_epoch000028.pth"
os.environ["VGGT_DIR"] = hf_hub_download("Yuxihenry/SpatialTrackerCkpts", "spatrack_front.pth") #, force_download=True)
VGGT_DIR = os.environ["VGGT_DIR"]
assert os.path.exists(VGGT_DIR), f"VGGT_DIR {VGGT_DIR} does not exist"
front_track = VGGT4Track()