43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
import logging
|
|
from pathlib import Path
|
|
from src.runner import run
|
|
from src.config import presets
|
|
|
|
|
|
def main():
|
|
import argparse
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
|
|
)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-b", "--base_path", help="Base path", default="output")
|
|
parser.add_argument(
|
|
"-v", "--video_path", help="Video path", default="example/video.mp4"
|
|
)
|
|
parser.add_argument(
|
|
"-o",
|
|
"--output",
|
|
help="Output video name (example: 'interpolated_video.mp4')",
|
|
default="interpolated_video.mp4",
|
|
)
|
|
parser.add_argument(
|
|
"-p",
|
|
"--preset",
|
|
help="Model preset",
|
|
choices=["small", "large", "global"],
|
|
default="global",
|
|
)
|
|
args = parser.parse_args()
|
|
run(
|
|
base_path=Path(args.base_path),
|
|
video_path=Path(args.video_path),
|
|
output_video=args.output,
|
|
preset=getattr(presets, args.preset.upper()),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|