File size: 940 Bytes
57746f1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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)
|