mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2025-04-05 13:35:00 +00:00
13 lines
396 B
Python
13 lines
396 B
Python
from diff_gaussian_rasterization import compute_relocation
|
|
import torch
|
|
import math
|
|
|
|
N_max = 51
|
|
binoms = torch.zeros((N_max, N_max)).float().cuda()
|
|
for n in range(N_max):
|
|
for k in range(n+1):
|
|
binoms[n, k] = math.comb(n, k)
|
|
|
|
def compute_relocation_cuda(opacity_old, scale_old, N):
|
|
N.clamp_(min=1, max=N_max-1)
|
|
return compute_relocation(opacity_old, scale_old, N, binoms, N_max) |