Skip to content

Commit

Permalink
refactor: remove debug logs from nerfstudio and add info logs
Browse files Browse the repository at this point in the history
  • Loading branch information
YifuTao committed Oct 7, 2024
1 parent d76a023 commit e485ba0
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion scripts/reconstruction_benchmark/nerf.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,8 @@ def update_argv(nerfstudio_config):


def run_nerfstudio(ns_config):
logger.info(f"Running '{ns_config['method']}' on {ns_config['data']}")
logging.disable(logging.DEBUG)
update_argv(ns_config)
train_entrypoint()
sys.argv = [sys.argv[0]]
Expand All @@ -75,8 +77,10 @@ def run_nerfstudio(ns_config):
latest_output_config = lastest_output_folder / "config.yml"

# evaluate renders
logger.info(f"Evaluating from {lastest_output_folder}")
render_dir = lastest_output_folder / "renders"
run_nerfstudio_eval(latest_output_config, render_dir)
logging.disable(logging.NOTSET)

# export cloud
export_method = "gaussian-splat" if ns_config["method"] == "splatfacto" else "pointcloud"
Expand All @@ -99,7 +103,7 @@ def run_nerfstudio_eval(config_file, render_dir):
}
update_argv(eval_config)
eval_entrypoint()
logger.info(f"Eval results: {json.load(output_eval_file.open())}")
logger.info(f"Nerfstudio eval results\n{json.load(output_eval_file.open())}")
sys.argv = [sys.argv[0]]


Expand Down

0 comments on commit e485ba0

Please sign in to comment.