Skip to content

Commit

Permalink
automated 3DGS insertion; added to bib
Browse files Browse the repository at this point in the history
  • Loading branch information
myla committed Feb 10, 2025
1 parent 3cc5870 commit be4d578
Show file tree
Hide file tree
Showing 3 changed files with 52 additions and 36 deletions.
70 changes: 35 additions & 35 deletions data_extraction/build_html.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
"TanksAndTemples": ["23.14", "0.841", "0.183", 411000000, 1783867.00],
"MipNeRF360": ["27.21", "0.815", "0.214", 734000000, 3362470.00],
"DeepBlending": ["29.41", "0.903", "0.243", 676000000, 2975634.50],
"SyntheticNeRF": ["33.32", None, None, None, None],
"SyntheticNeRF": ["33.31", None, None, None, None],
}

colors = [
Expand Down Expand Up @@ -153,32 +153,32 @@ def combine_tables_to_html():
# filter out "Baseline" keyword from Submethod
df["Submethod"] = df["Submethod"].str.replace("Baseline", "")

# insert 3DGS
df = pd.concat(
[
df,
pd.DataFrame(
[
{
"Method": "3DGS",
"Submethod": "",
"PSNR": org_3dgs[dataset][0],
"SSIM": (
org_3dgs[dataset][1] if org_3dgs[dataset][1] else ""
),
"LPIPS": (
org_3dgs[dataset][2] if org_3dgs[dataset][1] else ""
),
"Size [Bytes]": org_3dgs[dataset][3],
"#Gaussians": org_3dgs[dataset][4],
"Data Source": "",
"Comment": "",
}
]
),
],
ignore_index=True,
)
# # insert 3DGS
# df = pd.concat(
# [
# df,
# pd.DataFrame(
# [
# {
# "Method": "3DGS",
# "Submethod": "",
# "PSNR": org_3dgs[dataset][0],
# "SSIM": (
# org_3dgs[dataset][1] if org_3dgs[dataset][1] else ""
# ),
# "LPIPS": (
# org_3dgs[dataset][2] if org_3dgs[dataset][1] else ""
# ),
# "Size [Bytes]": org_3dgs[dataset][3],
# "#Gaussians": org_3dgs[dataset][4],
# "Data Source": "",
# "Comment": "",
# }
# ]
# ),
# ],
# ignore_index=True,
# )

# parse all float columns to float and keep the exact numer of decimal places
df["PSNR"] = df["PSNR"].apply(lambda x: Decimal(x) if x != "" else None)
Expand Down Expand Up @@ -505,7 +505,7 @@ def add_top_3_classes(df, actual_df):
return actual_df

def add_top_3_classes_per_type(
df, actual_df, category_col="category", include_3dgs=True
df, actual_df, category_col="category", include_3dgs=False
):
colors = ["first", "second", "third"]
for col in df.columns:
Expand All @@ -523,12 +523,12 @@ def add_top_3_classes_per_type(
filtered_indices = df[df[category_col] == category_type].index
filtered_col = float_col[filtered_indices]

# Explicitly include the 3DGS-30K row in the filtered column
if include_3dgs:
filtered_indices = filtered_indices.union(
df[df["Method"].str.contains("3DGS-30K", na=False)].index
)
filtered_col = float_col[filtered_indices]
# # Explicitly include the 3DGS-30K row in the filtered column
# if include_3dgs:
# filtered_indices = filtered_indices.union(
# df[df["Method"].str.contains("3DGS-30K", na=False)].index
# )
# filtered_col = float_col[filtered_indices]

if (
any(
Expand Down Expand Up @@ -985,7 +985,7 @@ def get_plot_data(ranks):

checkbox_states = {}
for method in shortnames.values():
if method in ["Scaffold-GS", "AtomGS", "GaussianPro"]:
if method in ["3DGS-30K", "Scaffold-GS", "AtomGS", "GaussianPro"]:
checkbox_states[method] = False
else:
checkbox_states[method] = True
Expand Down
6 changes: 5 additions & 1 deletion data_extraction/data_source.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -65,4 +65,8 @@ liu2024compgs:

lee2025compression3dgaussiansplatting:
url: "https://github.com/w-m/3dgs-compression-survey/tree/main/sources/results_lee2025compression3dgaussiansplatting"
is_csv: True
is_csv: True

kerbl3Dgaussians:
url: "https://github.com/w-m/3dgs-compression-survey/tree/main/sources/results_kerbl20233dgs"
is_csv: True
12 changes: 12 additions & 0 deletions methods_compression.bib
Original file line number Diff line number Diff line change
Expand Up @@ -196,4 +196,16 @@ @misc{lee2025compression3dgaussiansplatting
primaryClass={cs.CV},
url={https://arxiv.org/abs/2501.03399},
shortname={CodecGS},
}

@article{kerbl3Dgaussians,
author = {Kerbl, Bernhard and Kopanas, Georgios and Leimk{\"u}hler, Thomas and Drettakis, George},
title = {3D Gaussian Splatting for Real-Time Radiance Field Rendering},
journal = {ACM Transactions on Graphics},
number = {4},
volume = {42},
month = {July},
year = {2023},
url = {https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/},
shortname={3DGS-30K},
}

0 comments on commit be4d578

Please sign in to comment.