From e5a8f45d72cc470ecc180fcf5fb615f2c3c19842 Mon Sep 17 00:00:00 2001 From: hvalev Date: Tue, 3 Oct 2023 19:10:38 +0200 Subject: [PATCH] 9x speed improvement to generateNoiseImage --- R/generateNoiseImage.R | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/R/generateNoiseImage.R b/R/generateNoiseImage.R index eccf2cd..55a8fa2 100644 --- a/R/generateNoiseImage.R +++ b/R/generateNoiseImage.R @@ -30,7 +30,9 @@ generateNoiseImage <- function(params, p) { p <- list(patches=p$sinusoids, patchIdx=p$sinIdx, noise_type='sinusoid') } - noise <- apply(p$patches * array(params[p$patchIdx], dim(p$patches)), 1:2, mean) + patch_indices <- p$patchIdx + patch_params <- array(params[patch_indices], dim(p$patches)) + reshaped_matrix <- array(p$patches * patch_params, dim(p$patches)) + noise <- array(rowMeans(reshaped_matrix), dim(p$patches)[1:2]) return(noise) - } \ No newline at end of file