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

Clean fft.py

Browse Source
This commit is contained in:
Benjamin Loison 2024-05-14 00:24:02 +02:00
parent 451bd55219
commit 4bab354a95
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8
1 changed files with 25 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
datasets/raise/fft/fft.py
View File

@ -2,35 +2,40 @@ from scipy import fftpack
import numpy as np
import imageio
from PIL import Image, ImageDraw
import matplotlib.pyplot as plt
image1 = imageio.imread('image.jpg',as_gray=True)
image = imageio.imread('image.jpg', mode = 'L')
#plt.imshow(image, cmap = 'grey')
#plt.show()
#convert image to numpy array
image1_np=np.array(image1)
# convert image to numpy array
imageNp = np.array(image)
#fft of image
fft1 = fftpack.fftshift(fftpack.fft2(image1_np))
# fft of image
fft1 = fftpack.fftshift(fftpack.fft2(imageNp))
#Create a low pass filter image
x,y = image1_np.shape[0],image1_np.shape[1]
#size of circle
e_x,e_y=50,50
#create a box
bbox=((x/2)-(e_x/2),(y/2)-(e_y/2),(x/2)+(e_x/2),(y/2)+(e_y/2))
# create a low pass filter image
x, y = imageNp.shape
# size of circle
CIRCLE_RADIUS = 50
eX, eY = CIRCLE_RADIUS, CIRCLE_RADIUS
# create a box
middleX, middleY = x / 2, y / 2
bbox = (middleX - eX, middleY - eY, middleX + eX, middleY + eY)
low_pass=Image.new("L",(image1_np.shape[0],image1_np.shape[1]),color=0)
lowPass = Image.new('L', imageNp.shape, color = 0)
draw1=ImageDraw.Draw(low_pass)
draw1.ellipse(bbox, fill=1)
draw = ImageDraw.Draw(lowPass)
draw.ellipse(bbox, fill = 1)
low_pass_np=np.array(low_pass)
lowPassNp = np.array(lowPass)
#multiply both the images
filtered=np.multiply(fft1,low_pass_np)
# multiply both the images
filtered = np.multiply(fft1, lowPassNp.T)
#inverse fft
# inverse fft
ifft2 = np.real(fftpack.ifft2(fftpack.ifftshift(filtered)))
ifft2 = np.maximum(0, np.minimum(ifft2, 255))
#save the image
imageio.imsave('fft-then-ifft.png', ifft2.astype(np .uint8))
# save the image
imageio.imsave('fft-then-ifft.png', ifft2.astype(np.uint8))