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

Correct a typo but PRNU classifier still looking random

Browse Source
This commit is contained in:
Benjamin Loison 2024-03-22 12:51:34 +01:00
parent 36f5b69f5e
commit 41926663b7
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
datasets/fake/generate_dataset.py
View File

@ -22,9 +22,9 @@ PRNU_FACTOR = 0.1
def randomImage(scale):
return np.maximum(np.random.normal(loc = 0, scale = scale, size = (IMAGE_SIZE, IMAGE_SIZE)), 0)
prnus = [randomImage(scale = 0.1) for _ in range(NUMBER_OF_PHONES)]
prnus = [randomImage(scale = PRNU_FACTOR) for _ in range(NUMBER_OF_PHONES)]
images = [[randomImage(scale = 10) + prnus[phoneIndex] for _ in range(NUMBER_OF_IMAGES_PER_PHONE)] for phoneIndex in range(NUMBER_OF_PHONES)]
images = [[randomImage(scale = 1) + prnus[phoneIndex] for _ in range(NUMBER_OF_IMAGES_PER_PHONE)] for phoneIndex in range(NUMBER_OF_PHONES)]
# How to simplify?
allImages = np.max([np.max(prnus) + np.max(images)])
@ -49,8 +49,9 @@ def getPhoneIndexByNearestPrnu(estimatedPrnu):
if nearestPrnuRmsDiff is None or prnuRmsDiff < nearestPrnuRmsDiff:
nearestPrnuRmsDiff = prnuRmsDiff
nearestPrnuPhoneIndex = phoneIndex
return phoneIndex
return nearestPrnuPhoneIndex
# Should choose between *average* and *mean*, I would prefer *mean*.
# What about CAI on `phoneImagesMean`?
correctGuessesByAverageImages = 0
correctGuessesByAverageCAIImages = 0
@ -65,13 +66,10 @@ for phoneIndex in range(NUMBER_OF_PHONES):
print('RMS diff with average CAI images =', rmsdiff(caiImagesMean, phonePrnu))
guessedPhoneIndexByAverageImages = getPhoneIndexByNearestPrnu(phoneImagesMean)
guessedPhoneIndexByAverageCAIImages = getPhoneIndexByNearestPrnu(caiImagesMean)
#print(f'Actual phone index {phoneIndex}, guessed phone index {guessedPhoneIndex}')
print(f'Actual phone index {phoneIndex}, guessed phone index {guessedPhoneIndexByAverageImages} by averages images and {guessedPhoneIndexByAverageCAIImages} by average CAI images')
correctGuessesByAverageImages += 1 if phoneIndex == guessedPhoneIndexByAverageImages else 0
correctGuessesByAverageCAIImages += 1 if phoneIndex == guessedPhoneIndexByAverageCAIImages else 0
#correct = phoneIndex == guessedPhoneIndex
#if correct:
# correctGuesses += 1
print(correctGuessesByAverageImages / NUMBER_OF_PHONES)
print(correctGuessesByAverageCAIImages / NUMBER_OF_PHONES)
print(f'{correctGuessesByAverageImages / NUMBER_OF_PHONES=}')
print(f'{correctGuessesByAverageCAIImages / NUMBER_OF_PHONES=}')