INNER CODE UNIT · Python
main
MIT-SPARK/VGGT-SLAM · main.py:37
def main():
"""
Main function that wraps the entire pipeline of VGGT-SLAM.
"""
args = parser.parse_args()
use_optical_flow_downsample = True
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"Using device: {device}")
solver = Solver(
init_conf_threshold=args.conf_threshold,
lc_thres=args.lc_thres,
vis_voxel_size=args.vis_voxel_size,
vis_imgs=args.vis_imgs
)
print("Initializing and loading VGGT model...")