diff --git a/notebooks/fit_model_to_scene_full.ipynb b/notebooks/fit_model_to_scene_full.ipynb index 09e6323..45df7d9 100644 --- a/notebooks/fit_model_to_scene_full.ipynb +++ b/notebooks/fit_model_to_scene_full.ipynb @@ -130,16 +130,21 @@ "outputs": [], "source": [ "PATH_TO_VIDEO = \"../assets/examples/video_fruits.mp4\"\n", + "perform_preprocessing = True\n", + "\n", "num_ref_views = 16 # how many frames you want to extract from video and colmap\n", - "if False:\n", + "num_corrs = 20000\n", + "\n", + "if perform_preprocessing:\n", " # Full pipeline helpers\n", + " import shutil \n", " from gradio_demo import preprocess_input\n", " from source.utils_preprocess import read_video_frames, preprocess_frames, select_optimal_frames, save_frames_to_scene_dir, run_colmap_on_scene\n", " from source.visualization import generate_circular_camera_path, save_numpy_frames_as_mp4, generate_fully_smooth_cameras_with_tsp, put_text_on_image\n", " # Change to your path\n", " images, scene_dir = preprocess_input(PATH_TO_VIDEO, num_ref_views, num_corrs)\n", - " shutil.copytree(scene_dir, '../assets/video_colmaped', dirs_exist_ok=True)\n", - " \n" + " os.makedirs('.../assets/video_colmaped', exist_ok=True)\n", + " shutil.copytree(scene_dir, '../assets/video_colmaped', dirs_exist_ok=True)" ] }, {