From d553a6528d4a3443d837d296c43bdb730c7dd57c Mon Sep 17 00:00:00 2001 From: mirjagranfors Date: Mon, 15 Sep 2025 12:01:20 +0200 Subject: [PATCH] Update noises --- deeptrack/noises.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/deeptrack/noises.py b/deeptrack/noises.py index eaaa5afe1..cce4bc123 100644 --- a/deeptrack/noises.py +++ b/deeptrack/noises.py @@ -217,7 +217,11 @@ def get( # For a Torch backend. elif self.get_backend() == "torch": - noisy_image = mu + image + torch.randn(*image.shape) * sigma + noisy_image = ( + mu + + image + + torch.randn(*image.shape, device=image.device) * sigma + ) return noisy_image @@ -300,8 +304,8 @@ def get( # For a Torch backend. elif self.get_backend() == "torch": - real_noise = torch.randn(*image.shape) - imag_noise = torch.randn(*image.shape) * 1j + real_noise = torch.randn(*image.shape, device=image.device) + imag_noise = torch.randn(*image.shape, device=image.device) * 1j noisy_image = mu + image + (real_noise + imag_noise) * sigma return noisy_image