mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 11:54:01 +08:00
cleanup
This commit is contained in:
1 parent
5cacc91142
commit
4acfce84c6
2 files changed
-45
No files matched your search
@@ -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))
|
||||
Reference in new issue
Block a user