From 671a692114657e67f29486ddc221ca9f67be57f0 Mon Sep 17 00:00:00 2001 From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com> Date: Tue, 30 Apr 2024 01:04:33 +0200 Subject: [PATCH] #59: Save `rmss` for futher investigations For instance to plot a fair comparison, see [issues/59#issuecomment-1732](https://gitea.lemnoslife.com/Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones/issues/59#issuecomment-1732). --- datasets/raise/extract_noise.py | 6 +----- datasets/raise/split_and_compare_prnus_of_subgroups.py | 1 + datasets/raise/utils.py | 6 +++++- 3 files changed, 7 insertions(+), 6 deletions(-) diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 7399576..1db2b78 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -4,7 +4,7 @@ import numpy as np import os from tqdm import tqdm import csv -from utils import Color, denoise, iterativeMean, isARawImage, escapeFilePath, getColorChannel +from utils import Color, denoise, iterativeMean, isARawImage, escapeFilePath, getColorChannel, saveNpArray import matplotlib.pyplot as plt from scipy.ndimage import gaussian_filter @@ -95,10 +95,6 @@ if (minColor is None or maxColor is None) and denoiser != 'mean': print(f'{minColor=}') print(f'{maxColor=}') -def saveNpArray(fileName, npArray): - with open(f'{fileName}.npy', 'wb') as f: - np.save(f, npArray) - if denoiser == 'mean': means = {} for color in colors: diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py index 6b07460..cab3a16 100755 --- a/datasets/raise/split_and_compare_prnus_of_subgroups.py +++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py @@ -56,5 +56,6 @@ plt.title(f'RMS between both subgroups estimated PRNUs with {DENOISER} denoiser plt.xlabel('Number of images of each subgroup') plt.ylabel('RMS between both subgroups estimated PRNUs') plt.plot(rmss) +saveNpArray('{setting}_rmss', rmss) plt.savefig(f'{setting}_rms_between_estimated_prnu_of_2_subgroups.svg') #plt.show() diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index cd74a25..825c73b 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -119,4 +119,8 @@ def mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages newX, newY = [getNewIndex(index, offset) for index, offset in [(x, offsetX), (y, offsetY)]] multipleColorsImage[newX, newY] = pixel - return multipleColorsImage \ No newline at end of file + return multipleColorsImage + +def saveNpArray(fileName, npArray): + with open(f'{fileName}.npy', 'wb') as f: + np.save(f, npArray)