From c2cf139261a3c731be957de5ede460e1b69fa511 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Tue, 11 Jun 2024 17:53:34 +0200 Subject: [PATCH] Correct `pearsonr` `import` --- datasets/raise/utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index 87c379d..98668c3 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -10,7 +10,7 @@ import builtins as __builtin__ from scipy.ndimage import gaussian_filter import os from skimage.restoration import estimate_sigma -from scipy import fftpack +from scipy import fftpack, stats import imageio class Color(Enum): @@ -44,7 +44,7 @@ class Distance(Enum): return self.name.replace('_', '').title() def getPearsonCorrelationDistance(firstImage, secondImage): - pearsonCorrelation = scipy.stats.pearsonr(firstImage.flatten(), secondImage.flatten()).statistic + pearsonCorrelation = stats.pearsonr(firstImage.flatten(), secondImage.flatten()).statistic if pearsonCorrelation < 0: print('Negative Pearson correlation, investigate further!') exit(1) @@ -161,7 +161,6 @@ def getColorChannel(imageFilePath, color): def escapeFilePath(filePath): return filePath.replace('/', '_') - def getNewIndex(index, offset): newIndex = (index - offset) * 2 + offset return newIndex