Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones
1
0
Fork 0
Code Issues 69 Pull Requests Packages Projects Releases 1 Wiki Activity

Add necessary denoiser argument

Browse Source
This commit is contained in:
Benjamin Loison
2024-05-03 02:45:27 +02:00
parent 82b14a8724
commit 9d33abcea3
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
datasets/raise/utils.py
View File

@@ -132,18 +132,18 @@ def updateExtremes(imageNpArray, minColor, maxColor):
def print(*toPrint):
__builtin__.print(datetime.now(), *toPrint)
def getColorMeans(imagesFileNames, colors = Color):
def getColorMeans(imagesFileNames, colors, denoiser):
colorMeans = {}
for color in colors:
colorIterativeMean = iterativeMean()
for imageFileName in tqdm(imagesFileNames, f'Computing mean of {color} colored images'):
imageNpArray = getImageNpArray(imageFileName, False, color)
imageNpArray = getImageNpArray(imageFileName, False, color, denoiser)
imageNpArray = gaussian_filter(imageNpArray, sigma = 5)
colorIterativeMean.add(imageNpArray)
colorMeans[color] = colorIterativeMean.mean
return colorMeans
def getImageNpArray(imageFilePath, computeExtremes, color):
def getImageNpArray(imageFilePath, computeExtremes, color, denoiser):
global minColor, maxColor
imageNpArray = getColorChannel(imageFilePath, color)