tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

commands

Browse Source
This commit is contained in:
Ethan Weber 2022-12-16 19:26:46 +00:00
parent c24ecf5d2a
commit 55f334d33b
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/datasets/process_mipnerf360.py
View File

@ -55,6 +55,8 @@ for split, gpu in zip(splits, gpus):
commands.append(command)
import subprocess
import sys
def f(cmd):
out = subprocess.run(cmd, capture_output=False, shell=True, check=False)
if out.returncode != 0:
@ -63,9 +65,18 @@ def f(cmd):
if out.stdout is not None:
return out.stdout.decode("utf-8")
from multiprocessing import Pool
with Pool() as p:
print(p.map(f, commands))
# evaluation...
#
# the default method
# DATASET="person" METHOD="nerfacto-default" && ns-train nerfacto --vis wandb --data "data/nerfstudio/${DATASET}" --wandb-name "${METHOD}-${DATASET}" --experiment-name "${METHOD}-${DATASET}"
# without pose optimization
# DATASET="person" METHOD="nerfacto-no-pose" && ns-train nerfacto --vis wandb --data "data/nerfstudio/${DATASET}" --wandb-name "${METHOD}-${DATASET}" --experiment-name "${METHOD}-${DATASET}" --pipeline.datamanager.camera-optimizer.mode off