strusov/JaxPM_highres
0
0
Fork 0
forked from Aquila-Consortium/JaxPM_highres
Code Pull requests Activity

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