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

Add gaussianNoise to getPrnuShownAsSuch but applies on the whole PRNU image

Browse Source
This commit is contained in:
Benjamin Loison 2024-03-29 13:16:50 +01:00
parent 2d96bdc225
commit ec7c5c6688
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
algorithms/image_utils/image_utils.py
View File

@ -16,13 +16,13 @@ def showImageWithMatplotlib(npArray, title = None, cmap = 'viridis'):
def toPilImage(npArray):
return Image.fromarray(npArray)
def getPrnuShownAsSuch(size):
def getPrnuShownAsSuch(size, gaussianNoise = 0):
# Supports `WIDTH` > `HEIGHT` and conversely.
WIDTH, HEIGHT = size
TEXT = 'PRNU'
image = Image.new('L', size)
draw = ImageDraw.Draw(image)
imagePil = Image.new('L', size)
draw = ImageDraw.Draw(imagePil)
fontPath = os.path.expanduser('~/.local/share/fonts/impact.ttf')
for fontSize in range(1, HEIGHT + 1):
font = ImageFont.truetype(fontPath, fontSize)
@ -30,4 +30,5 @@ def getPrnuShownAsSuch(size):
break
# Center vertically, especially in the case `HEIGHT` > `WIDTH`.
draw.text((0, HEIGHT // 2 - fontSize // 2), TEXT, 255, font = font)
return np.array(image)
imageNpArray = np.array(imagePil)
return imageNpArray + randomGaussianImage(gaussianNoise, size[::-1])

5
datasets/noise_free_test_images/estimate_prnu.py
View File

@ -32,7 +32,10 @@ IMAGE_SIZE_SHAPE = [dimension // SPLIT_N_X_N for dimension in (704, 469)]
np.random.seed(0)
#prnuNpArray = 255 * randomGaussianImage(scale = PRNU_FACTOR, size = IMAGE_SIZE_SHAPE)
prnuNpArray = getPrnuShownAsSuch(IMAGE_SIZE_SHAPE) * PRNU_FACTOR
prnuNpArray = getPrnuShownAsSuch(IMAGE_SIZE_SHAPE, 100) * PRNU_FACTOR
showImageWithMatplotlib(prnuNpArray)
##
def isIn256Range(x):
return 0 <= x and x <= 255