BiohazardVFX/4DGaussians
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix stage rendering error.

Browse Source
This commit is contained in:
Geralt_of_Rivia 2024-03-03 15:27:41 +08:00 committed by GitHub
parent f4054ef962
commit e2e0f999c8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
gaussian_renderer/__init__.py
View File

@ -77,9 +77,9 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
scales = pc._scaling
rotations = pc._rotation
deformation_point = pc._deformation_table
if stage == "coarse" :
if "coarse" in stage:
means3D_final, scales_final, rotations_final, opacity_final, shs_final = means3D, scales, rotations, opacity, shs
else:
elif "fine" in stage:
# time0 = get_time()
# means3D_deform, scales_deform, rotations_deform, opacity_deform = pc._deformation(means3D[deformation_point], scales[deformation_point],
# rotations[deformation_point], opacity[deformation_point],
@ -87,9 +87,8 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
means3D_final, scales_final, rotations_final, opacity_final, shs_final = pc._deformation(means3D, scales,
rotations, opacity, shs,
time)
# time1 = get_time()
# print("deformation forward:",time1-time0)
# print(time.max())
else:
raise NotImplementedError