diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py index 6fc5e44..5e385bb 100755 --- a/datasets/raise/split_and_compare_prnus_of_subgroups.py +++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py @@ -3,7 +3,7 @@ import numpy as np import matplotlib.pyplot as plt from tqdm import tqdm -from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color, mergeSingleColorChannelImagesAccordingToBayerFilter, rescaleRawImageForDenoiser, updateExtremes +from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color, mergeSingleColorChannelImagesAccordingToBayerFilter, rescaleRawImageForDenoiser, updateExtremes, saveNpArray import sys import os from random import shuffle @@ -67,6 +67,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) +saveNpArray(f'{setting}_rmss', rmss) plt.savefig(f'{setting}_rms_between_estimated_prnu_of_2_subgroups.svg') #plt.show()