Fix PSF used for reconstruction debug examples.

LCAV · Oct 4, 2024 · fc89d28 · fc89d28
1 parent 5d3dbb0
commit fc89d28
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/recon/train_learning_based.py b/scripts/recon/train_learning_based.py
@@ -397,7 +397,7 @@ def train_learned(config):
                     i,
                     lensed,
                     lensless,
-                    psf,
+                    psf_recon,
                     test_set,
                     "",
                     flip_lr,
@@ -415,7 +415,7 @@ def train_learned(config):
                         i,
                         lensed,
                         (lensless - background),
-                        psf,
+                        psf_recon,
                         test_set,
                         "subtraction_",
                         flip_lr,