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

Simplify and clean split_and_compare_prnus_of_subgroups.py

Browse Source
This commit is contained in:
Benjamin Loison
2024-04-27 16:06:47 +02:00
parent ee7353cfde
commit e9dc28dc2a
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
datasets/raise/split_and_compare_prnus_of_subgroups.py
View File

@@ -5,6 +5,7 @@ from tqdm import tqdm
from utils import denoise from utils import denoise
from skimage import img_as_float from skimage import img_as_float
import sys import sys
import os
sys.path.insert(0, '../../algorithms/distance/') sys.path.insert(0, '../../algorithms/distance/')
@@ -12,19 +13,18 @@ from rms_diff import rmsDiffNumpy
NUMBER_OF_SUBGROUPS = 1 NUMBER_OF_SUBGROUPS = 1
DENOISER = 'wavelet' DENOISER = 'wavelet'
IMAGES_FOLDER = 'flat-field/TIF' IMAGES_FOLDER = 'flat-field/TIF'
imagesFileNames = os.listdir(IMAGES_FOLDER) imagesFileNames = os.listdir(IMAGES_FOLDER)
numberOfImagesPerSubgroup = len(imagesFileNames) // NUMBER_OF_SUBGROUPS numberOfImagesPerSubgroup = len(imagesFileNames) // NUMBER_OF_SUBGROUPS
# 1
numberOfImagesThresholds = range(numberOfImagesPerSubgroup, numberOfImagesPerSubgroup + 1) numberOfImagesThresholds = range(numberOfImagesPerSubgroup, numberOfImagesPerSubgroup + 1)
# Assume random image order to not introduce a bias. # Assume random image order to not introduce a bias.
subgroupsPrnuEstimatesNpArray = [] subgroupsPrnuEstimatesNpArray = []
for subgroupIndex in range(NUMBER_OF_SUBGROUPS): for subgroupIndex in tqdm(range(NUMBER_OF_SUBGROUPS), 'Subgroup'):
imagesPrnuEstimateNpArray = [] imagesPrnuEstimateNpArray = []
for imageFileName in tqdm(imagesFileNames[numberOfImagesPerSubgroup * subgroupIndex : numberOfImagesPerSubgroup * (subgroupIndex + 1)]): subgroupImagesFileNames = imagesFileNames[numberOfImagesPerSubgroup * subgroupIndex : numberOfImagesPerSubgroup * (subgroupIndex + 1)]
for imageFileName in tqdm(subgroupImagesFileNames, f'Image of subgroup {subgroupIndex}'):
imagePath = f'{IMAGES_FOLDER}/{imageFileName}' imagePath = f'{IMAGES_FOLDER}/{imageFileName}'
imagePil = Image.open(imagePath) imagePil = Image.open(imagePath)
imageNpArray = img_as_float(np.array(imagePil)) imageNpArray = img_as_float(np.array(imagePil))
@@ -39,7 +39,7 @@ for subgroupIndex in range(NUMBER_OF_SUBGROUPS):
subgroupsPrnuEstimatesNpArray += [subgroupPrnuEstimateNpArray] subgroupsPrnuEstimatesNpArray += [subgroupPrnuEstimateNpArray]
rmss = [] rmss = []
for numberOfImagesIndex, numberOfImages in enumerate(numberOfImagesThresholds): for numberOfImagesIndex in range(len(numberOfImagesThresholds)):
rms = rmsDiffNumpy(subgroupsPrnuEstimatesNpArray[0][numberOfImagesIndex], subgroupsPrnuEstimatesNpArray[1][numberOfImagesIndex]) rms = rmsDiffNumpy(subgroupsPrnuEstimatesNpArray[0][numberOfImagesIndex], subgroupsPrnuEstimatesNpArray[1][numberOfImagesIndex])
rmss += [rms] rmss += [rms]