Added human readable formatting

This commit is contained in:
origamiman72 committed 2022-07-21 21:47:05 -07:00
1 parent 8cfe141fd7
commit 66d23199ff
2 files changed
+11 -1

No files matched your search

+2 -1
View File
@@ -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