From ff2ab887c21ca2385030b366adfa70464827e502 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Thu, 26 Jan 2023 21:59:07 +0000 Subject: [PATCH] benchmarks --- nerfstudio/data/datamanagers/base_datamanager.py | 2 ++ scripts/benchmarking/benchmark_nerfstudio_paper.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index 555d618e4..d4742efd4 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -45,6 +45,7 @@ from nerfstudio.data.dataparsers.instant_ngp_dataparser import ( InstantNGPDataParserConfig, ) from nerfstudio.data.dataparsers.nerfstudio_dataparser import NerfstudioDataParserConfig +from nerfstudio.data.dataparsers.mipnerf360_dataparser import Mipnerf360DataParserConfig from nerfstudio.data.dataparsers.nuscenes_dataparser import NuScenesDataParserConfig from nerfstudio.data.dataparsers.phototourism_dataparser import ( PhototourismDataParserConfig, @@ -63,6 +64,7 @@ AnnotatedDataParserUnion = tyro.conf.OmitSubcommandPrefixes[ # Omit prefixes of tyro.extras.subcommand_type_from_defaults( { "nerfstudio-data": NerfstudioDataParserConfig(), + "mipnerf360-data": Mipnerf360DataParserConfig(), "blender-data": BlenderDataParserConfig(), "friends-data": FriendsDataParserConfig(), "instant-ngp-data": InstantNGPDataParserConfig(), diff --git a/scripts/benchmarking/benchmark_nerfstudio_paper.py b/scripts/benchmarking/benchmark_nerfstudio_paper.py index 66ef7f436..362114fd3 100644 --- a/scripts/benchmarking/benchmark_nerfstudio_paper.py +++ b/scripts/benchmarking/benchmark_nerfstudio_paper.py @@ -20,7 +20,7 @@ mipnerf360_table_rows = [ ( "nerfacto-w/o-pose-app", "nerfacto", - "--pipeline.eval_optimize_cameras False --pipeline.eval_optimize_appearance False --pipeline.datamanager.camera-optimizer.mode off --pipeline.model.use-appearance-embedding False nerfstudio-data --downscale-factor 4 --train-split-percentage 0.875", + "--pipeline.eval_optimize_cameras False --pipeline.eval_optimize_appearance False --pipeline.datamanager.camera-optimizer.mode off --pipeline.model.use-appearance-embedding False nerfstudio-data --downscale-factor 4 --train-split-percentage 0.875 mipnerf360-data", ), ]