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

Add and use getImageFilePath

Browse Source
This commit is contained in:
Benjamin Loison 2024-05-03 02:27:07 +02:00
parent 514413e1dd
commit 790c0d6da4
No known key found for this signature in database
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
datasets/raise/extract_noise.py
View File

@ -46,13 +46,17 @@ if type_ is not None:
minColor = None
maxColor = None
def getImageNpArray(imageFileName, computeExtremes, color):
global minColor, maxColor
def getImageFilePath(imageFileName):
if raiseNotFlatFields:
imageFileName = imageFileName.replace('.png', '.NEF')
imageFilePath = f'{imagesFolderPath}/nef/{imageFileName}'
else:
imageFilePath = f'{imagesFolderPath}/{imageFileName}'
return imageFilePath
def getImageNpArray(imageFileName, computeExtremes, color):
global minColor, maxColor
imageFilePath = getImageFilePath(imageFileName)
imageNpArray = getColorChannel(imageFilePath, color)
if computeExtremes: