"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "e1144ac20c4c96f09f1c1e203715969bf80feb65"
Commit 224e217d authored by Ruilong Li's avatar Ruilong Li
Browse files

renaming

parent de4b66a1
instant-ngp @ 25dec33c
Subproject commit 25dec33c253f035485bb2e1f8563e12ef3134e8b
from ._backend import _C from ._backend import _C
ray_marching = _C.ray_marching volumetric_marching = _C.volumetric_marching
volumetric_rendering_steps = _C.volumetric_rendering_steps volumetric_rendering_steps = _C.volumetric_rendering_steps
...@@ -30,7 +30,7 @@ torch::Tensor volumetric_rendering_weights_backward( ...@@ -30,7 +30,7 @@ torch::Tensor volumetric_rendering_weights_backward(
torch::Tensor sigmas torch::Tensor sigmas
); );
std::vector<torch::Tensor> ray_marching( std::vector<torch::Tensor> volumetric_marching(
// rays // rays
const torch::Tensor rays_o, const torch::Tensor rays_o,
const torch::Tensor rays_d, const torch::Tensor rays_d,
...@@ -47,7 +47,7 @@ std::vector<torch::Tensor> ray_marching( ...@@ -47,7 +47,7 @@ std::vector<torch::Tensor> ray_marching(
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{ {
m.def("ray_aabb_intersect", &ray_aabb_intersect); m.def("ray_aabb_intersect", &ray_aabb_intersect);
m.def("ray_marching", &ray_marching); m.def("volumetric_marching", &volumetric_marching);
m.def("volumetric_rendering_steps", &volumetric_rendering_steps); m.def("volumetric_rendering_steps", &volumetric_rendering_steps);
m.def("volumetric_rendering_weights_forward", &volumetric_rendering_weights_forward); m.def("volumetric_rendering_weights_forward", &volumetric_rendering_weights_forward);
m.def("volumetric_rendering_weights_backward", &volumetric_rendering_weights_backward); m.def("volumetric_rendering_weights_backward", &volumetric_rendering_weights_backward);
......
...@@ -212,7 +212,7 @@ __global__ void marching_forward_kernel( ...@@ -212,7 +212,7 @@ __global__ void marching_forward_kernel(
} }
std::vector<torch::Tensor> ray_marching( std::vector<torch::Tensor> volumetric_marching(
// rays // rays
const torch::Tensor rays_o, const torch::Tensor rays_o,
const torch::Tensor rays_d, const torch::Tensor rays_d,
......
...@@ -4,7 +4,7 @@ from typing import Callable, Tuple ...@@ -4,7 +4,7 @@ from typing import Callable, Tuple
import torch import torch
from .cuda import ( # ComputeWeight,; VolumeRenderer,; ray_aabb_intersect, from .cuda import ( # ComputeWeight,; VolumeRenderer,; ray_aabb_intersect,
ray_marching, volumetric_marching,
volumetric_rendering_steps, volumetric_rendering_steps,
) )
from .utils import ( from .utils import (
...@@ -55,7 +55,7 @@ def volumetric_rendering( ...@@ -55,7 +55,7 @@ def volumetric_rendering(
frustum_dirs, frustum_dirs,
frustum_starts, frustum_starts,
frustum_ends, frustum_ends,
) = ray_marching( ) = volumetric_marching(
# rays # rays
rays_o, rays_o,
rays_d, rays_d,
......
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