彩色图像的简单变换

import cv2
import numpy as np
from matplotlib import pyplot as plt
import random
import math

平移

img = cv2.imread('../image/lena-color.jpg',-1)

a = 200
b = 300
img1 = np.zeros((round(img.shape[0]),round(img.shape[1]),3),dtype='uint8')

for i in range(img1.shape[0]):
    for j in range(img1.shape[1]):
        if round(i + a) < img.shape[0] and round(j + b) < img.shape[1]:
            img1[i+a][j+b] = img[i][j]

b,g,r = cv2.split(img)
img3 = cv2.merge([r, g, b])
b,g,r = cv2.split(img1)
img1 = cv2.merge([r, g, b])

plt.subplot(121),plt.imshow(img3)
plt.title('input Image'), plt.xticks([]), plt.yticks([])
plt.subplot(122),plt.imshow(img1)
plt.title('move Image'), plt.xticks([]), plt.yticks([])
plt.show()

旋转

img = cv2.imread('../image/lena-color.jpg',-1)


img2 = np.zeros((round(img.shape[0]),round(img.shape[1]),3),dtype='uint8')
for i in range(img2.shape[0]):
    for j in range(img2.shape[1]):
        if img.shape[0]-i< img.shape[0]:
            img2[img.shape[0]-i][j] = img[i][j]

b,g,r = cv2.split(img)
img1 = cv2.merge([r, g, b])
b,g,r = cv2.split(img2)
img2 = cv2.merge([r, g, b])

plt.subplot(121),plt.imshow(img1)
plt.title('input Image'), plt.xticks([]), plt.yticks([])
plt.subplot(122),plt.imshow(img2)
plt.title('flip Image'), plt.xticks([]), plt.yticks([])
plt.show()

缩放

img = cv2.imread('../image/lena-color.jpg',-1)

a = 0.5
b = 2
img2 = np.zeros((round(img.shape[0]*a),round(img.shape[1]*b),3),dtype='uint8')

for i in range(img2.shape[0]):
    for j in range(img2.shape[1]):
        if round(i/a) < img.shape[0] and round(j/b) < img.shape[1]:
            img2[i][j] = img[round(i/a)][round(j/b)]

b,g,r = cv2.split(img)
img1 = cv2.merge([r, g, b])
b,g,r = cv2.split(img2)
img2 = cv2.merge([r, g, b])

plt.subplot(121),plt.imshow(img1)
plt.title('input Image'), plt.xticks([]), plt.yticks([])
plt.subplot(122),plt.imshow(img2)
plt.title('zoom Image'), plt.xticks([]), plt.yticks([])
plt.show()

傅里叶变换

img = cv2.imread('../image/lena-color.jpg',0)

dft = cv2.dft(np.float32(img), flags=cv2.DFT_COMPLEX_OUTPUT)
dft_shift = np.fft.fftshift(dft)

magnitude_spectrum = 20 * np.log(cv2.magnitude(dft_shift[:, :, 0], dft_shift[:, :, 1]))

plt.subplot(121), plt.imshow(img)
plt.title('Input Image'), plt.xticks([]), plt.yticks([])
plt.subplot(122), plt.imshow(magnitude_spectrum)
plt.title('Magnitude Spectrum'), plt.xticks([]), plt.yticks([])
plt.show()

简单的低通滤波

import cv2
import numpy as np
import matplotlib.pyplot as plt

kernel_25h = np.array([[-1, -1, -1, -1, -1],
                      [-1, 1, 2, 1, -1],
                      [-1, 2, 4, 2, -1],
                      [-1, 1, 2, 1, -1],
                      [-1, -1, -1, -1, -1]])

kernel_25l = np.array([[0.04, 0.04, 0.04, 0.04, 0.04],
                      [0.04, 0.04, 0.04, 0.04, 0.04],
                      [0.04, 0.04, 0.04, 0.04, 0.04],
                      [0.04, 0.04, 0.04, 0.04, 0.04],
                      [0.04, 0.04, 0.04, 0.04, 0.04]])

img = cv2.imread('../image/lena-color.jpg',-1)
b,g,r = cv2.split(img)
img = cv2.merge([r, g, b])
ndimg = np.array(img)

k3 = cv2.filter2D(ndimg, -1, kernel_25h)
k5 = cv2.filter2D(ndimg, -1, kernel_25l)

plt.subplot(131)
plt.imshow(img)
plt.title("source image")

plt.subplot(132)
plt.imshow(k3)
plt.title("kernel5hpf")

plt.subplot(133)
plt.imshow(k5)
plt.title("kernel5lpf")
plt.show()