diff --git a/nerfstudio/process_data/video_to_nerfstudio_dataset.py b/nerfstudio/process_data/video_to_nerfstudio_dataset.py index 928752e43..af17e7d6b 100644 --- a/nerfstudio/process_data/video_to_nerfstudio_dataset.py +++ b/nerfstudio/process_data/video_to_nerfstudio_dataset.py @@ -16,6 +16,7 @@ import shutil from dataclasses import dataclass +from typing import Literal from nerfstudio.process_data import equirect_utils, process_data_utils from nerfstudio.process_data.colmap_converter_to_nerfstudio_dataset import ColmapConverterToNerfstudioDataset @@ -36,6 +37,10 @@ class VideoToNerfstudioDataset(ColmapConverterToNerfstudioDataset): """Target number of frames to use per video, results may not be exact.""" percent_radius_crop: float = 1.0 """Create circle crop mask. The radius is the percent of the image diagonal.""" + matching_method: Literal["exhaustive", "sequential", "vocab_tree"] = "sequential" + """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.""" def main(self) -> None: """Process video into a nerfstudio dataset."""