guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-02-22 17:47:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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)
# We compute the value of the filter at frequency 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