From 28ba7d759bdf74d343f10053477ca6d274780805 Mon Sep 17 00:00:00 2001 From: Anthony-Tafoya Date: Wed, 11 Sep 2024 23:13:06 +0000 Subject: [PATCH] NERF-0000: Adding Random Seed Frame Processing --- nerfstudio/process_data/process_data_utils.py | 1 + nerfstudio/process_data/video_to_nerfstudio_dataset.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/nerfstudio/process_data/process_data_utils.py b/nerfstudio/process_data/process_data_utils.py index 5001acdc816..fb2914711bf 100644 --- a/nerfstudio/process_data/process_data_utils.py +++ b/nerfstudio/process_data/process_data_utils.py @@ -15,6 +15,7 @@ """Helper utils for processing data into the nerfstudio format.""" import math +import random import re import shutil import sys diff --git a/nerfstudio/process_data/video_to_nerfstudio_dataset.py b/nerfstudio/process_data/video_to_nerfstudio_dataset.py index e522f4a28ef..f7415bb6522 100644 --- a/nerfstudio/process_data/video_to_nerfstudio_dataset.py +++ b/nerfstudio/process_data/video_to_nerfstudio_dataset.py @@ -41,9 +41,9 @@ class VideoToNerfstudioDataset(ColmapConverterToNerfstudioDataset): """Feature matching method to use. Vocab tree is recommended for a balance of speed and accuracy. Exhaustive is slower but more accurate. Sequential is faster but should only be used for videos.""" - random_seed: int = None + random_seed: Optional[int] = None """Random seed to select video frames""" - eval_random_seed: int = None + eval_random_seed: Optional[int] = None """Random seed to select video frames for eval set""" def main(self) -> None: