From 30db00689dffb53a43dbe9d036dd35017e0129ac Mon Sep 17 00:00:00 2001 From: julien-sahli Date: Thu, 11 May 2023 12:46:08 +0200 Subject: [PATCH] fixed indentation error in gradient descent inital estimation --- lensless/gradient_descent.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/lensless/gradient_descent.py b/lensless/gradient_descent.py index 33b88374..1bc7434c 100644 --- a/lensless/gradient_descent.py +++ b/lensless/gradient_descent.py @@ -115,10 +115,9 @@ def reset(self): self._image_est = np.ones(self._psf_shape, dtype=self._dtype) else: assert self._initial_est.shape == self._psf_shape - - psf_flat = self._psf.reshape(-1, self._psf_shape[3]) - pixel_start = (np.max(psf_flat, axis=0) + np.min(psf_flat, axis=0)) / 2 - self._image_est *= pixel_start + psf_flat = self._psf.reshape(-1, self._psf_shape[3]) + pixel_start = (np.max(psf_flat, axis=0) + np.min(psf_flat, axis=0)) / 2 + self._image_est *= pixel_start # set step size as < 2 / lipschitz Hadj_flat = self._convolver._Hadj.reshape(-1, self._psf_shape[3])