mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 19:33:46 +08:00
Added human readable formatting
This commit is contained in:
1 parent
8cfe141fd7
commit
66d23199ff
2 files changed
+11
-1
No files matched your search
+2
-1
@@ -15,6 +15,7 @@ from tqdm import tqdm
|
||||
from pyrad.cameras.camera_paths import CameraPath, get_interpolated_camera_path, get_spiral_path
|
||||
from pyrad.data.dataloader import EvalDataloader, setup_dataset_eval, setup_dataset_train
|
||||
from pyrad.graphs.base import Graph, setup_graph
|
||||
from pyrad.utils.misc import human_format
|
||||
from pyrad.utils.writer import TimeWriter
|
||||
|
||||
|
||||
@@ -200,7 +201,7 @@ def main(
|
||||
avg_rays_per_sec = stats_dict["avg rays per sec"]
|
||||
avg_fps = stats_dict["avg fps"]
|
||||
print(f"Avg. PSNR: {avg_psnr:0.4f}")
|
||||
print(f"Avg. Rays / Sec: {avg_rays_per_sec:.2e}")
|
||||
print(f"Avg. Rays / Sec: {human_format(avg_rays_per_sec)}")
|
||||
print(f"Avg. FPS: {avg_fps:0.4f}")
|
||||
elif method == MethodType.TRAJ:
|
||||
# TODO(ethan): pass in camera information into argparse parser
|
||||
|
||||
Reference in new issue
Block a user