29 lines
669 B
Python
29 lines
669 B
Python
import cv2
|
|
import numpy as np
|
|
|
|
car = cv2.imread("../../data/car.png", cv2.IMREAD_GRAYSCALE)
|
|
dog = cv2.imread("../../data/dog.png", cv2.IMREAD_GRAYSCALE)
|
|
|
|
''' FFT '''
|
|
CAR = np.fft.fft2(car)
|
|
CAR_magnitude = np.abs(CAR)
|
|
CAR_angle = np.angle(CAR)
|
|
|
|
DOG = np.fft.fft2(dog)
|
|
DOG_magnitude = np.abs(DOG)
|
|
DOG_angle = np.angle(DOG)
|
|
|
|
mix1 = CAR_magnitude * np.exp(1j * DOG_angle)
|
|
mix2 = DOG_magnitude * np.exp(1j * CAR_angle)
|
|
|
|
mix1 = np.fft.ifft2(mix1).astype(np.float32)
|
|
mix2 = np.fft.ifft2(mix2).astype(np.float32)
|
|
|
|
''' Bild anzeigen '''
|
|
cv2.imshow("car", car)
|
|
cv2.imshow("dog", dog)
|
|
cv2.imshow("mix1", mix1 / np.max(mix1))
|
|
cv2.imshow("mix2", mix2 / np.max(mix2))
|
|
|
|
cv2.waitKey()
|