Finish: ALL of lantern-nerf

nerfstudio/process_data/equirect_utils.py

@@ -252,7 +252,7 @@ def generate_planar_projections_from_equirectangular(
 
     return output_dir, output_mask_dir, output_hdr_dir
 
-def generate_planar_projections_from_equirectangular_2(
+def generate_planar_projections_from_equirectangular_GT(
     metadata_path: Path,
     image_dir: Path,
     planar_image_size: Tuple[int, int],

nerfstudio/scripts/viewer/run_viewer.py

@@ -21,6 +21,7 @@ from __future__ import annotations
 import time
 from dataclasses import dataclass, field, fields
 from pathlib import Path
+
 from typing import Literal
 
 import tyro
@@ -112,9 +113,8 @@ def _start_viewer(config: TrainerConfig, pipeline: Pipeline, step: int):
 
     assert viewer_state and pipeline.datamanager.train_dataset
     viewer_state.init_scene(
-        train_dataset=pipeline.datamanager.train_dataset,
+        dataset=pipeline.datamanager.train_dataset,
         train_state="completed",
-        eval_dataset=pipeline.datamanager.eval_dataset,
     )
     if isinstance(viewer_state, ViewerState):
         viewer_state.viser_server.set_training_state("completed")
@@ -133,4 +133,4 @@ if __name__ == "__main__":
     entrypoint()
 
 # For sphinx docs
-get_parser_fn = lambda: tyro.extras.get_parser(RunViewer)  # noqa
+get_parser_fn = lambda: tyro.extras.get_parser(RunViewer)  # noqa

nerfstudio/viewer/server/viewer_state.py

@@ -440,4 +440,4 @@ class ViewerState:
     def training_complete(self) -> None:
         """Called when training is complete."""
         self.training_state = "completed"
-        self.viser_server.set_training_state("completed")
+        self.viser_server.set_training_state("completed")