tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

Merge branch 'gs-export-2' of github.com:nerfstudio-project/nerfstudio into gs-export-2

Browse Source
This commit is contained in:
akristoffersen 2023-10-30 06:26:07 +00:00
commit c64f86fe3c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nerfstudio/scripts/exporter.py
View File

@ -505,10 +505,10 @@ class ExportGaussianSplat(Exporter):
colors = model.colors_all.data.cpu().numpy()
colors_diffuse = colors[:, 0, ..., np.newaxis]
colors = colors[:, 0, ..., np.newaxis]
shs = colors[:, 1:].reshape((colors.shape[0], -1, 1))
for i in range(colors_diffuse.shape[1]):
for i in range(colors.shape[1]):
if i < 3:
map_to_tensors[f"f_dc_{i}"] = colors[:, i]
for i in range(shs.shape[1]):