tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

update benchmarking; add clip to mipnerf lpips

Browse Source
This commit is contained in:
evonneng 2022-07-07 18:59:23 +00:00
parent d0f0f8da2d
commit 39b67e6fe9
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyrad/graphs/mipnerf.py
View File

@ -173,7 +173,7 @@ class MipNerfGraph(Graph):
coarse_psnr = self.psnr(image, rgb_coarse)
fine_psnr = self.psnr(image, rgb_fine)
fine_ssim = self.ssim(image, rgb_fine)
fine_lpips = self.lpips(image, rgb_fine)
fine_lpips = self.lpips(image, torch.clip(rgb_fine, min=-1, max=1))
writer.put_scalar(name=f"psnr/val_{image_idx}-coarse", scalar=float(coarse_psnr), step=step)
writer.put_scalar(name=f"psnr/val_{image_idx}-fine", scalar=float(fine_psnr), step=step)

6
scripts/benchmarking/run_benchmark.py
View File

@ -55,8 +55,6 @@ def _load_hydra_config(hydra_dir: str) -> DictConfig:
hydra_dir = f"{hydra_dir}/{basename}"
initialize(config_path=os.path.join("../../", hydra_dir, ".hydra/"))
config = compose("config.yaml")
## set the tensorboard to a /tmp/ output file for now
config.logging.writer.TensorboardWriter.log_dir = "/tmp/"
return config
@ -81,7 +79,7 @@ def main(args):
"meta info": {"graph": args.graph, "benchmark_date": args.benchmark_date, "hydra": args.hydra_base_dir},
"results": benchmarks,
}
json_file = os.path.join(args.hydra_base_dir, f"{args.timestamp}.json")
json_file = os.path.join(args.hydra_base_dir, f"{args.benchmark_date}.json")
with open(json_file, "w", encoding="utf8") as f:
json.dump(benchmark_info, f, indent=2)
logging.info("saved benchmark results to %s", json_file)
@ -91,6 +89,6 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-g", "--graph", type=str, required=True, help="name of nerf graph to benchmark")
parser.add_argument("-d", "--benchmark_date", type=str, required=True, help="timestamp of run to benchmark")
parser.add_argument("-h", "--hydra_base_dir", type=str, required=True, help="hydra base output path")
parser.add_argument("-o", "--hydra_base_dir", type=str, default="outputs/", help="hydra base output path")
args = parser.parse_args()
main(args)