From 2d3043c41bcb587bf8b3b0baa987849fca8eac5b Mon Sep 17 00:00:00 2001 From: Daewoon Kim <84174755+wooni-github@users.noreply.github.com> Date: Mon, 11 Mar 2024 19:20:25 +0900 Subject: [PATCH] update Camera models in data processing script (#2997) * update colmap camera models --- .../process_data/colmap_converter_to_nerfstudio_dataset.py | 2 +- nerfstudio/process_data/process_data_utils.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py index 924c4b752d..35d45b9050 100644 --- a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py +++ b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py @@ -29,7 +29,7 @@ class ColmapConverterToNerfstudioDataset(BaseConverterToNerfstudioDataset): """Base class to process images or video into a nerfstudio dataset using colmap""" - camera_type: Literal["perspective", "fisheye", "equirectangular"] = "perspective" + camera_type: Literal["perspective", "fisheye", "equirectangular", "pinhole", "simple_pinhole"] = "perspective" """Camera model to use.""" matching_method: Literal["exhaustive", "sequential", "vocab_tree"] = "vocab_tree" """Feature matching method to use. Vocab tree is recommended for a balance of speed diff --git a/nerfstudio/process_data/process_data_utils.py b/nerfstudio/process_data/process_data_utils.py index 468781b775..d754525d37 100644 --- a/nerfstudio/process_data/process_data_utils.py +++ b/nerfstudio/process_data/process_data_utils.py @@ -49,12 +49,16 @@ class CameraModel(Enum): OPENCV = "OPENCV" OPENCV_FISHEYE = "OPENCV_FISHEYE" EQUIRECTANGULAR = "EQUIRECTANGULAR" + PINHOLE = "PINHOLE" + SIMPLE_PINHOLE = "SIMPLE_PINHOLE" CAMERA_MODELS = { "perspective": CameraModel.OPENCV, "fisheye": CameraModel.OPENCV_FISHEYE, "equirectangular": CameraModel.EQUIRECTANGULAR, + "pinhole": CameraModel.PINHOLE, + "simple_pinhole": CameraModel.SIMPLE_PINHOLE, }