Skip to content

Commit c9f1d7e

Browse files
author
Felix Igelbrink
committed
added missing members to denoiser and fixed example
1 parent 4ca66c1 commit c9f1d7e

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

examples/denolser.py renamed to examples/denoiser.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -59,10 +59,10 @@
5959
ret = cp.asnumpy(ret)
6060

6161
fig, axs = plt.subplots(nrows=2, sharex=True, sharey=True)
62-
axs[0].imshow(np.clip(color_image, 0, 255).astype(np.uint8))
62+
axs[0].imshow(np.clip(color_image, 0, 1).astype(np.float32))
6363
axs[0].set_title("original")
6464

65-
ret = np.clip(ret, 0, 255).astype(np.uint8)
65+
ret = np.clip(ret, 0, 1).astype(np.float32)
6666
axs[1].imshow(ret)
6767
axs[1].set_title("denoised")
6868
plt.show()

optix/denoiser.pxd

+3
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,9 @@ cdef class Denoiser(OptixContextObject):
191191
cdef size_t _state_size
192192
cdef object _d_scratch
193193
cdef size_t _scratch_size
194+
cdef size_t _guide_layer_scratch_size
195+
cdef size_t _intensity_scratch_size
196+
cdef size_t _average_color_scratch_size
194197
cdef object _d_window
195198
cdef size_t _window_size
196199
cdef object _d_intensity

optix/denoiser.pyx

+1
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ cdef class Denoiser(OptixContextObject):
152152
self._scratch_size = 0
153153

154154
self._guide_layer_scratch_size = 0
155+
self._intensity_scratch_size = 0
155156
self._average_color_scratch_size = 0
156157
self._state_size = 0
157158

0 commit comments

Comments
 (0)