tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Switch default matching method for videos to 'sequential' in COLMAP (#2912)

Browse Source 
default matching method for video to sequential for COLMAP
This commit is contained in:
Justin Kerr 2024-02-13 12:26:17 -08:00 committed by GitHub
parent eb9bcb577d
commit 7f9def1ff0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nerfstudio/process_data/video_to_nerfstudio_dataset.py
View File

@ -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."""