app.py

import argparse
from slam3r.app.app_i2p import run_i2p


def get_args_parser():
    parser = argparse.ArgumentParser(description="A demo for SLAM3R's I2P model")
    parser.add_argument("--device", type=str, 
        # default='cuda', 
        default='cpu',
        help="pytorch device")
    parser.add_argument("--tmp_dir", type=str, default="./_tmp", 
        help="value for tempfile.tempdir")
    return parser


if __name__ == "__main__":
    parser = get_args_parser()
    args = parser.parse_args()
    run_i2p(parser)