Add missing import for saveNpArray and add a missing f-string

This commit is contained in:
Benjamin Loison 2024-04-30 03:35:30 +02:00
parent c2a2dea2c7
commit ad5aa22ae6
No known key found for this signature in database

View File

@ -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()