mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
1 parent
8af3a7a385
commit
cf34ef4bd0
26 files changed
+384
-355
No files matched your search
@@ -59,7 +59,7 @@ for dataset in ${DATASETS[@]}; do
|
||||
export CUDA_VISIBLE_DEVICES=${GPU_IDX[$idx]}
|
||||
base_config_name=${config_name/"graph_"/""}
|
||||
config_path="${output_dir}/blender_${dataset}_${month}-${date}-${year}/${base_config_name}/${year}-${month}-${date}_${seconds}/nerfactory_models/"
|
||||
python scripts/run_eval.py compute-psnr \
|
||||
python scripts/eval.py compute-psnr \
|
||||
--load-config=${config_path} \
|
||||
--output-path=${output_dir}/${base_config_name}/blender_${dataset}_${month}-${date}-${year}_${seconds}.json &
|
||||
echo "Launched ${checkpoint_dir} on gpu ${GPU_IDX[$idx]}"
|
||||
|
||||
@@ -52,7 +52,7 @@ len=${#GPU_IDX[@]}
|
||||
|
||||
for dataset in ${DATASETS[@]}; do
|
||||
export CUDA_VISIBLE_DEVICES=${GPU_IDX[$idx]}
|
||||
python scripts/run_train.py \
|
||||
python scripts/train.py \
|
||||
${config_name} \
|
||||
--logging.writer.2.no-enable \
|
||||
--pipeline.datamanager.train-dataset.data-directory=data/blender/${dataset} \
|
||||
|
||||
@@ -237,14 +237,19 @@ def main(
|
||||
CONSOLE.print()
|
||||
CONSOLE.print("Notes:")
|
||||
CONSOLE.print(
|
||||
":warning: In bash, completions will trigger via [dim white]./scripts/run_train.py <TAB>[/dim white], but"
|
||||
" not [dim white]python ./scripts/run_train.py <TAB>[/dim white]."
|
||||
":warning: In bash, completions will trigger via [dim white]./scripts/train.py <TAB>[/dim white], but"
|
||||
" not [dim white]python ./scripts/train.py <TAB>[/dim white]."
|
||||
)
|
||||
CONSOLE.print(
|
||||
":warning: Completions are in an experimental state. If you run into any issues, please file an issue!"
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
def entrypoint():
|
||||
"""Entrypoint for use with pyproject scripts."""
|
||||
dcargs.extras.set_accent_color("bright_yellow")
|
||||
dcargs.cli(main, description=__doc__)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
entrypoint()
|
||||
@@ -1,6 +1,6 @@
|
||||
#!/usr/bin/env python
|
||||
"""
|
||||
run_eval.py
|
||||
eval.py
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
@@ -197,7 +197,8 @@ class RenderTrajectory:
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
def entrypoint():
|
||||
"""Entrypoint for use with pyproject scripts."""
|
||||
# A Union over dataclass types will create a subcommand for each type.
|
||||
#
|
||||
# TODO: it would make sense to split this script up into separate scripts.
|
||||
@@ -208,3 +209,7 @@ if __name__ == "__main__":
|
||||
# checkpoint when resuming a training run.
|
||||
dcargs.extras.set_accent_color("bright_yellow")
|
||||
dcargs.cli(Union[ComputePSNR, RenderTrajectory]).main()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
entrypoint()
|
||||
@@ -482,6 +482,11 @@ def main(
|
||||
CONSOLE.rule()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
def entrypoint():
|
||||
"""Entrypoint for use with pyproject scripts."""
|
||||
dcargs.extras.set_accent_color("bright_yellow")
|
||||
dcargs.cli(main)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
entrypoint()
|
||||
@@ -1,6 +1,6 @@
|
||||
#!/usr/bin/env python
|
||||
"""
|
||||
run_train_nerf.py
|
||||
train.py
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
@@ -204,7 +204,12 @@ def main(config: cfg.Config) -> None:
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
def entrypoint():
|
||||
"""Entrypoint for use with pyproject scripts."""
|
||||
# Choose a base configuration and override values.
|
||||
dcargs.extras.set_accent_color("bright_yellow")
|
||||
main(dcargs.cli(AnnotatedBaseConfigUnion))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
entrypoint()
|
||||
Reference in new issue
Block a user