4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
This commit is contained in:
Ethan Weber 2023-01-11 02:19:44 +00:00
parent 5cacc91142
commit 4acfce84c6
2 changed files with 0 additions and 45 deletions

View File

@ -1,36 +0,0 @@
"""
Code to create the tables.
"""
import pandas as pd
# from nerfstudio.scripts.benchmarking.benchmark_nerfacto import table_rows
# the csv filenames
csv_filename = "wandb-psnr.csv"
# code to read the csv file
df = pd.read_csv(csv_filename)
column_titles = list(df.columns.values)
last_row = list(df.values[-1])
print(len(column_titles))
print(len(last_row))
from collections import defaultdict
results = defaultdict(dict)
for column_title, last_row_value in zip(column_titles, last_row):
if column_title == "Step" or "__MAX" in column_title or "__MIN" in column_title:
continue
name = column_title.split(" - ")[0]
capture_name, method = name.split("_")
metric = float(last_row_value)
# format to have 2 decimal places
metric = f"{metric:.2f}"
results[capture_name][method] = metric
import pprint
pprint.pprint(results)
print(len(results))

View File

@ -145,17 +145,8 @@ nerfstudio_file_ids = {
"stump": "https://data.nerf.studio/nerfstudio/stump.zip",
"sculpture": "https://data.nerf.studio/nerfstudio/sculpture.zip",
"Giannini-Hall": "https://data.nerf.studio/nerfstudio/Giannini-Hall.zip",
# MipNeRF-360 Dataset. TODO: add these links
"mipnerf360-bicycle": "",
"mipnerf360-garden": "",
"mipnerf360-stump": "",
"mipnerf360-room": "",
"mipnerf360-counter": "",
"mipnerf360-kitchen": "",
"mipnerf360-bonsai": "",
"all": None,
"nerfstudio-dataset": nerfstudio_dataset,
"mipnerf360-dataset": mipnerf360_dataset,
}
NerfstudioCaptureName = tyro.extras.literal_type_from_choices(nerfstudio_file_ids.keys())