From e97801951ac6d414ee568f33a6a9e68c7ff97b9c Mon Sep 17 00:00:00 2001 From: Vikram Voleti Date: Mon, 18 Mar 2024 17:13:42 +0000 Subject: [PATCH] Minor updates --- README.md | 8 +++++++- scripts/sampling/simple_video_sample.py | 2 ++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 44a4ced3..dd9d61d8 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,13 @@ - SV3D_u: This variant generates orbital videos based on single image inputs without camera conditioning.. - SV3D_p: Extending the capability of SVD3_u, this variant accommodates both single images and orbital views allowing for the creation of 3D video along specified camera paths. - We extend the streamlit demo `scripts/demo/video_sampling.py` and the standalone python script `scripts/sampling/simple_video_sample.py` for inference of both models. - - Please check our [tech report](https://sv3d.github.io/static/paper.pdf) and [video summary](https://youtu.be/Zqw4-1LcfWg) for more details. + - Please check our [project page](https://sv3d.github.io), [tech report](https://sv3d.github.io/static/paper.pdf) and [video summary](https://youtu.be/Zqw4-1LcfWg) for more details. + +To run SV3D on a single image: +`python scripts/sampling/simple_video_sample.py --input_path --version sv3d_p` + +To run SVD or SV3D on a streamlit server: +`streamlit run scripts/demo/video_sampling.py` ![tile](assets/sv3d.gif) diff --git a/scripts/sampling/simple_video_sample.py b/scripts/sampling/simple_video_sample.py index bc64591e..6d34e7cd 100644 --- a/scripts/sampling/simple_video_sample.py +++ b/scripts/sampling/simple_video_sample.py @@ -1,9 +1,11 @@ import math import os +import sys from glob import glob from pathlib import Path from typing import List, Optional +sys.path.append(os.path.realpath(os.path.join(os.path.dirname(__file__), "../../"))) import cv2 import imageio import numpy as np