src/infer.py

import argparse
import sys

sys.path.append('.')
from src.model import LSM_MASt3R
from src.utils.visualization_utils import render_video_from_file

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--file_list', type=str, nargs='+', required=True,
                        help='List of input image files or directories')
    parser.add_argument('--model_path', type=str, required=True)
    parser.add_argument('--output_path', type=str, required=True)
    parser.add_argument('--resolution', type=int, default=512)
    parser.add_argument('--n_interp', type=int, default=90)
    parser.add_argument('--fps', type=int, default=30)

    args = parser.parse_args()
    
    # 1. load model
    model = LSM_MASt3R.from_pretrained(args.model_path)
    # 2. render video
    render_video_from_file(args.file_list, model, args.output_path, resolution=args.resolution, n_interp=args.n_interp, fps=args.fps)