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

first-commit

Browse Source
This commit is contained in:
xiaoyuxi 2025-07-08 16:37:13 +08:00
parent fd1a134647
commit 19ec398bcf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -46,7 +46,7 @@ if __name__ == "__main__":
VGGT_DIR = os.environ["VGGT_DIR"] VGGT_DIR = os.environ["VGGT_DIR"]
assert os.path.exists(VGGT_DIR), f"VGGT_DIR {VGGT_DIR} does not exist" assert os.path.exists(VGGT_DIR), f"VGGT_DIR {VGGT_DIR} does not exist"
front_track = VGGT4Track() front_track = VGGT4Track()
front_track.load_state_dict(torch.load(VGGT_DIR)["model"], strict=False) front_track.load_state_dict(torch.load(VGGT_DIR), strict=False)
front_track.eval() front_track.eval()
front_track = front_track.to("cuda") front_track = front_track.to("cuda")