mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 10:34:29 +08:00
cleanup
This commit is contained in:
parent
5cacc91142
commit
4acfce84c6
@ -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))
|
||||
@ -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())
|
||||
|
||||
Loading…
Reference in New Issue
Block a user