NERF-0000: Adding Random Seed Frame Processing

nerfstudio-project · Sep 11, 2024 · 28ba7d7 · 28ba7d7
1 parent acb4007
commit 28ba7d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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: