guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-02-23 01:57:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

small fix

Browse source
This commit is contained in:
EiffL 2022-05-17 23:49:12 +02:00
parent aff8db56f5
commit 60309e1490
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/utils.py
View file

@ -92,7 +92,7 @@ def gaussian_smoothing(im, sigma):
axis=-1) axis=-1)
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.pdf(k, 0, 1. / (2. * np.pi * sigma))
filter /= filter[0,0] filter /= filter[0,0]
return jnp.fft.ifft2(jnp.fft.fft2(im) * filter).real return jnp.fft.ifft2(jnp.fft.fft2(im) * filter).real