Aquila-Consortium/JaxPM_highres
0
0
Fork 1
forked from guilhem_lavaux/JaxPM
Code Pull requests Activity

minor correction to gaussian smoothing

Browse source
This commit is contained in:
EiffL 2022-05-17 23:02:01 +02:00
parent ca5f26c93f
commit 5dc239927f
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
jaxpm/utils.py
View file

@ -93,6 +93,7 @@ def gaussian_smoothing(im, sigma):
k = jnp.linalg.norm(kvec, axis=-1) k = jnp.linalg.norm(kvec, axis=-1)
# We compute the value of the filter at frequency k # We compute the value of the filter at frequency k
filter = norm(0, 1. / (2. * np.pi * sigma)).pdf(k) filter = norm(0, 1. / (2. * np.pi * sigma)).pdf(k)
filter /= filter[0,0]
return jnp.fft.ifft2(jnp.fft.fft2(im) * filter).real return jnp.fft.ifft2(jnp.fft.fft2(im) * filter).real