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

Clean datasets/raise/extract_noise.py

Browse Source
This commit is contained in:
Benjamin Loison 2024-04-15 16:07:11 +02:00
parent a09f2ba1c0
commit 0445d98f52
No known key found for this signature in database
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
datasets/raise/extract_noise.py
View File

@ -17,8 +17,6 @@ mean = None
numberOfImagesInMean = 0
import csv
import magic
import re
deviceFiles = {}
resolutionFiles = {}
@ -32,19 +30,13 @@ with open('RAISE_all.csv') as csvfile:
device = row['Device']
resolution = row['Image Size']
deviceFiles[file] = device
#resolutionFiles[file] = resolution
image = Image.open(f'{imagesFolderPath}/{file}')
#t = magic.from_file(f'{imagesFolderPath}/{file}.png')
resolution = image.size
resolutionFiles[file] = resolution#re.search('(\d+) x (\d+)', t).groups()
#print(device, resolution)
resolutionFiles[file] = resolution
#files[f'{file}.png'] = row
imagesFileNames = os.listdir(imagesFolderPath)
#print(len(imagesFileNames))
# imageFileName.replace('.png', '')
imagesFileNames = [imageFileName for imageFileName in imagesFileNames if deviceFiles[imageFileName] == 'Nikon D7000' and resolutionFiles[imageFileName] == (4946, 3278)]
#print(len(imagesFileNames))
def treatImage(imageFileName):
global mean, numberOfImagesInMean