mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 02:24:38 +08:00
d71c05f256
* allows bridge server to run on a different machine * added script * update version * starting docs * update docs and scripts * ngrok support * use tcp * moving webrtc outside bridge server. wop * lazy commit * lazy commit * running code. webrtc not working * lazy commit * better logic for webrtc connection * cleanup * todo comment * working webrtc * removed logging * fixing black * fix eslint * fix python version * fix readme * misc cleanup * small docs fix Co-authored-by: Ethan Weber <ethanweber@Ethans-MacBook-Pro.local>
57 lines
1.5 KiB
Python
Executable File
57 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
eval.py
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
import tyro
|
|
from rich.console import Console
|
|
|
|
from nerfstudio.utils.eval_utils import eval_setup
|
|
|
|
CONSOLE = Console(width=120)
|
|
|
|
|
|
@dataclass
|
|
class ComputePSNR:
|
|
"""Load a checkpoint, compute some PSNR metrics, and save to a JSON."""
|
|
|
|
# Path to config YAML file.
|
|
load_config: Path
|
|
# Name of the output file.
|
|
output_path: Path = Path("output.json")
|
|
|
|
def main(self) -> None:
|
|
"""Main function."""
|
|
config, pipeline, checkpoint_path = eval_setup(self.load_config)
|
|
|
|
assert self.output_path.suffix == ".json"
|
|
metrics_dict = pipeline.get_average_eval_image_metrics()
|
|
# save output to some file
|
|
self.output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
benchmark_info = {
|
|
"experiment_name": config.experiment_name,
|
|
"method_name": config.method_name,
|
|
"checkpoint": str(checkpoint_path),
|
|
"results": metrics_dict,
|
|
}
|
|
self.output_path.write_text(json.dumps(benchmark_info, indent=2), "utf8")
|
|
CONSOLE.print(f"Saved results to: {self.output_path}")
|
|
|
|
|
|
def entrypoint():
|
|
"""Entrypoint for use with pyproject scripts."""
|
|
tyro.extras.set_accent_color("bright_yellow")
|
|
tyro.cli(ComputePSNR).main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
entrypoint()
|
|
|
|
# For sphinx docs
|
|
get_parser_fn = lambda: tyro.extras.get_parser(ComputePSNR) # noqa
|