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

misc cleanup (#432)

Browse Source 
* misc cuda improvements. added cone angle as param

* wandb as default. better organization for experiments

* turn off tensorboard

* disable eval when using viewer

* remove comment

* launch.json
This commit is contained in:
Ethan Weber authored and GitHub committed 2022-09-15 00:16:41 -07:00
1 parent ef8c4fbe23
commit 8973039e65
21 files changed
+230 -111
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

scripts/run_eval.py
+2 -1
View File
@@ -86,7 +86,8 @@ def _render_stats_dict(pipeline: Pipeline) -> Dict[str, float]:
with torch.no_grad():
image_idx = int(camera_ray_bundle.camera_indices[0, 0])
outputs = pipeline.model.get_outputs_for_camera_ray_bundle(camera_ray_bundle)
psnr = pipeline.model.log_test_image_outputs(image_idx, step, batch, outputs)
metrics_dict = pipeline.model.log_test_image_outputs(image_idx, step, batch, outputs)
psnr = metrics_dict["psnr"]
avg_rays_per_sec = _update_avg(avg_rays_per_sec, camera_ray_bundle.origins.shape[0] / t.duration, step)
avg_psnr = _update_avg(avg_psnr, psnr, step)
avg_fps = _update_avg(avg_fps, 1 / t.duration, step)