fix problem with addPoissonNoise

addPoissonNoise now correctly handles input that is not "double"

fix problem with addPoissonNoise
addPoissonNoise now correctly handles input that is not "double"
27f98c87 · Leon Merten Lohse · 05c6125a · 27f98c87
Commit 27f98c87 authored Oct 09, 2019 by Leon Merten Lohse
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

functions/generators/addPoissonNoise.m functions/generators/addPoissonNoise.m +8 -1

No files found.
--- a/functions/generators/addPoissonNoise.m
+++ b/functions/generators/addPoissonNoise.m
@@ -58,8 +58,15 @@ if nargin < 2
    photonsPerPixel = 10000;
 end

+imNoisy = zeros(size(im), 'like', im);
+
+% imnoise needs double as input
+if ~isa(im, 'double')
+  im = double(im);
+end
+
 poissonFactor = 1e-12*photonsPerPixel;
-imNoisy = (1./poissonFactor) * imnoise(poissonFactor * im,'poisson');
+imNoisy(:) = (1./poissonFactor) * imnoise(poissonFactor * im,'poisson');

 end