flashdrr 0.5__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- flashdrr/__init__.py +1 -0
- flashdrr/rendering/__init__.py +36 -0
- flashdrr/rendering/raycast.py +902 -0
- flashdrr/rendering/triton_raycast.py +678 -0
- flashdrr/utils/__init__.py +2 -0
- flashdrr/utils/volume_utils.py +46 -0
- flashdrr-0.5.dist-info/METADATA +223 -0
- flashdrr-0.5.dist-info/RECORD +11 -0
- flashdrr-0.5.dist-info/WHEEL +5 -0
- flashdrr-0.5.dist-info/licenses/LICENSE +21 -0
- flashdrr-0.5.dist-info/top_level.txt +1 -0
flashdrr/__init__.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = '0.5'
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
"""Differentiable volume rendering for PyTorch.
|
|
2
|
+
|
|
3
|
+
The package exposes two renderers and a handful of camera / geometry helpers:
|
|
4
|
+
|
|
5
|
+
* :class:`VolumeRaycaster` - the main Python-loop differentiable raycaster
|
|
6
|
+
(optionally with the :class:`DepthAwareScatter` and :class:`ASPP` building
|
|
7
|
+
blocks for scatter-based DRR rendering).
|
|
8
|
+
* :class:`FusedVolumeRenderer` - a Triton-fused forward kernel for fast
|
|
9
|
+
backprop-free rendering on CUDA.
|
|
10
|
+
|
|
11
|
+
The camera helpers (:func:`get_view_mat`, :func:`get_vtk_view_mat`,
|
|
12
|
+
:func:`get_proj_mat`, :func:`get_random_carm_views`, ...) are the same ones
|
|
13
|
+
the renderers use internally and are re-exported for downstream code.
|
|
14
|
+
"""
|
|
15
|
+
|
|
16
|
+
from .raycast import (
|
|
17
|
+
ASPP,
|
|
18
|
+
DepthAwareScatter,
|
|
19
|
+
VolumeRaycaster,
|
|
20
|
+
carm_to_camera_params,
|
|
21
|
+
get_random_carm_views,
|
|
22
|
+
get_vtk_view_mat,
|
|
23
|
+
piecewise_linear_channelwise,
|
|
24
|
+
)
|
|
25
|
+
from .triton_raycast import FusedVolumeRenderer
|
|
26
|
+
|
|
27
|
+
__all__ = [
|
|
28
|
+
'ASPP',
|
|
29
|
+
'DepthAwareScatter',
|
|
30
|
+
'FusedVolumeRenderer',
|
|
31
|
+
'VolumeRaycaster',
|
|
32
|
+
'carm_to_camera_params',
|
|
33
|
+
'get_random_carm_views',
|
|
34
|
+
'get_vtk_view_mat',
|
|
35
|
+
'piecewise_linear_channelwise',
|
|
36
|
+
]
|