proper WDD normalization

py4dstem · Oct 20, 2024 · 428c4ed · 428c4ed
1 parent b697818
commit 428c4ed
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/py4DSTEM/process/phase/direct_ptychography.py b/py4DSTEM/process/phase/direct_ptychography.py
@@ -2038,9 +2038,7 @@ def wrapper_function(**kwargs):
  for (ind_x, ind_y), res in zip(np.ndindex((sx, sy)), flat_results):
  psi[ind_x, ind_y] = res
 
- normalization = (
- xp.abs(psi[0, 0]) / np.prod(self._intensities_shape) / wdd_probe_0
- )
+ normalization = xp.abs(psi[0, 0]) / sx / sy
  psi /= normalization
 
  self._object = xp.fft.ifft2(psi)