UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-21 07:47:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix training report, add missing dependency

Browse Source
This commit is contained in:
alanvinx 2024-10-29 19:24:26 +01:00
parent 9e1c2c6197
commit 3dc0b75595
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
environment.yml
View File

@ -15,5 +15,6 @@ dependencies:
- pip:
- submodules/diff-gaussian-rasterization
- submodules/simple-knn
- submodules/fused_ssim
- opencv-python
- joblib

2
train.py
View File

@ -155,7 +155,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi
progress_bar.close()
# Log and save
training_report(tb_writer, iteration, Ll1, loss, l1_loss, iter_start.elapsed_time(iter_end), testing_iterations, scene, render, (pipe, background, dataset.train_test_exp, SPARSE_ADAM_AVAILABLE), dataset.train_test_exp)
training_report(tb_writer, iteration, Ll1, loss, l1_loss, iter_start.elapsed_time(iter_end), testing_iterations, scene, render, (pipe, background, 1., SPARSE_ADAM_AVAILABLE, None, dataset.train_test_exp), dataset.train_test_exp)
if (iteration in saving_iterations):
print("\n[ITER {}] Saving Gaussians".format(iteration))
scene.save(iteration)