Commit 3090d7bd authored by Ruilong Li's avatar Ruilong Li
Browse files

rename to "proposal_resampling"

parent a74f7e7b
...@@ -4,7 +4,7 @@ import torch ...@@ -4,7 +4,7 @@ import torch
from .grid import Grid from .grid import Grid
from .intersection import ray_aabb_intersect from .intersection import ray_aabb_intersect
from .sampling import proposal_sampling_with_filter, sample_along_rays from .sampling import proposal_resampling, sample_along_rays
@torch.no_grad() @torch.no_grad()
...@@ -171,12 +171,7 @@ def ray_marching( ...@@ -171,12 +171,7 @@ def ray_marching(
grid=grid, grid=grid,
) )
( ray_indices, t_starts, t_ends, proposal_samples = proposal_resampling(
ray_indices,
t_starts,
t_ends,
proposal_samples,
) = proposal_sampling_with_filter(
t_starts=t_starts, t_starts=t_starts,
t_ends=t_ends, t_ends=t_ends,
ray_indices=ray_indices, ray_indices=ray_indices,
......
...@@ -99,7 +99,7 @@ def sample_along_rays( ...@@ -99,7 +99,7 @@ def sample_along_rays(
@torch.no_grad() @torch.no_grad()
def proposal_sampling_with_filter( def proposal_resampling(
t_starts: torch.Tensor, # [n_samples, 1] t_starts: torch.Tensor, # [n_samples, 1]
t_ends: torch.Tensor, # [n_samples, 1] t_ends: torch.Tensor, # [n_samples, 1]
ray_indices: torch.Tensor, # [n_samples,] ray_indices: torch.Tensor, # [n_samples,]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment