tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 12:23:46 +08:00
Code Activity

cleanup prints

Browse Source
This commit is contained in:
Ethan Weber committed 2022-12-18 21:08:21 +00:00
1 parent 68245153b5
commit d74a0f6ecd
3 files changed
+118 -20
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

scripts/benchmarking/benchmark_nerfacto.py
+52
View File
@@ -0,0 +1,52 @@
"""
Process the Mipnerf360 dataset.
"""
from nerfstudio.utils.scripts import run_command
capture_names = [
"Egypt",
"person",
"kitchen",
"plane",
"dozer",
"floating-tree",
"aspen",
"stump",
"sculpture",
"Giannini-Hall",
]
table_rows = [
("nerfacto", ""),
("w/o-pose", "--pipeline.datamanager.camera-optimizer.mode off"),
("w/o-app", "--pipeline.model.use-appearance-embedding False"),
(
"w/o-pose-app",
"--pipeline.datamanager.camera-optimizer.mode off --pipeline.model.use-appearance-embedding False",
),
(
"1-prop-network",
"--pipeline.model.num-proposal-samples-per-ray \"256\" --pipeline.model.num_proposal_iterations 1",
),
]
# 30K iterations
for capture_name in capture_names:
for table_row_name, table_row_command in table_rows:
# TODO: find an available GPU
gpu = 0
command = " ".join(
(
# f"CUDA_VISIBLE_DEVICES={gpu}",
f"ns-train nerfacto",
"--vis wandb",
f"--data data/nerfstudio/{capture_name}",
"--pipeline.eval_optimize_cameras True --pipeline.eval_optimize_appearance True",
f"--wandb-name {capture_name}-{table_row_name} --experiment-name {capture_name}-{table_row_name}",
table_row_command,
)
)
print(command)
scripts/datasets/process_mipnerf360.py
+35
View File
@@ -91,3 +91,38 @@ with Pool() as p:
# DATASET="plane" METHOD="nerfacto-pose" && ns-train nerfacto --vis wandb --data "data/nerfstudio/${DATASET}" --wandb-name "${METHOD}" --experiment-name "${METHOD}-${DATASET}" --trainer.load-dir outputs/nerfacto-no-pose-plane/nerfacto/2022-12-16_220419/nerfstudio_models --trainer.steps-per-eval-all-images 10 --pipeline.eval_optimize_cameras True --pipeline.eval_optimize_appearance False
# DATASET="plane" METHOD="nerfacto-pose-app" && ns-train nerfacto --vis wandb --data "data/nerfstudio/${DATASET}" --wandb-name "${METHOD}" --experiment-name "${METHOD}-${DATASET}" --trainer.load-dir outputs/nerfacto-no-pose-plane/nerfacto/2022-12-16_220419/nerfstudio_models --trainer.steps-per-eval-all-images 10 --pipeline.eval_optimize_cameras True --pipeline.eval_optimize_appearance True
# DATASET="plane" METHOD="nerfacto-app" && ns-train nerfacto --vis wandb --data "data/nerfstudio/${DATASET}" --wandb-name "${METHOD}" --experiment-name "${METHOD}-${DATASET}" --trainer.load-dir outputs/nerfacto-no-pose-plane/nerfacto/2022-12-16_220419/nerfstudio_models --trainer.steps-per-eval-all-images 10 --pipeline.eval_optimize_cameras False --pipeline.eval_optimize_appearance True
capture_names = [
"Egypt",
"person",
"kitchen",
"plane",
"dozer",
"floating-tree",
"aspen",
"stump",
"sculpture",
"Giannini-Hall",
]
table_row = [
("nerfacto", ""),
("w/o-pose", "--pipeline.datamanager.camera-optimizer.mode off"),
("w/o-app", "--pipeline.model.use-appearance-embedding False"),
(
"w/o-pose-app",
"--pipeline.datamanager.camera-optimizer.mode off --pipeline.model.use-appearance-embedding False",
),
("1-prop-network", "--pipeline.model.num_proposal_samples_per_ray (256,) --pipeline.model.num_proposal_iterations 1"),
]
# 30K iterations
command = """
CAPTURE_NAME="plane"
TABLE_ROW="nerfacto-app"
&&
ns-train nerfacto --vis wandb --data "data/nerfstudio/${CAPTURE_NAME}" --wandb-name "${DATASET}-${TABLE_ROW}" --experiment-name "${DATASET}-${TABLE_ROW}"
--pipeline.eval_optimize_cameras True --pipeline.eval_optimize_appearance True
--trainer.steps-per-eval-all-images 30000 --trainer.max-num-iterations 30001
"""