Python: PS 滤镜--旋转模糊

本文用 Python 实现 PS 滤镜中的旋转模糊，具体的算法原理和效果可以参考之前的博客：

http://blog.csdn.net/matrix_space/article/details/39211667

from skimage import img_as_float

import matplotlib.pyplot as plt

from skimage import io

import numpy as np

import numpy.matlib

file_name='D:/Visual Effects/PS Algorithm/4.jpg'

img=io.imread(file_name)

img = img_as_float(img)

img_out = img.copy()

row, col, channel = img.shape

xx = np.arange (col)

yy = np.arange (row)

x_mask = numpy.matlib.repmat (xx, row, 1)

y_mask = numpy.matlib.repmat (yy, col, 1)

y_mask = np.transpose(y_mask)

center_y = (row -1) / 2.0

center_x = (col -1) / 2.0

R = np.sqrt((x_mask - center_x) **2 + (y_mask - center_y) ** 2)

angle = np.arctan2(y_mask - center_y , x_mask - center_x)

Num = 20

arr = ( np.arange(Num) + 1 ) / 100.0

for i in range (row):

    for j in range (col):

        T_angle = angle[i, j] + arr

        new_x = R[i, j] * np.cos(T_angle) + center_x

        new_y = R[i, j] * np.sin(T_angle) + center_y

        int_x = new_x.astype(int)

        int_y = new_y.astype(int)

        int_x[int_x > col-1] = col - 1

        int_x[int_x < 0] = 0

        int_y[int_y < 0] = 0

        int_y[int_y > row -1] = row -1

        img_out[i,j,0] = img[int_y, int_x, 0].sum()/Num

        img_out[i,j,1] = img[int_y, int_x, 1].sum()/Num

        img_out[i,j,2] = img[int_y, int_x, 2].sum()/Num

plt.figure(1)

plt.imshow(img)

plt.axis('off')

plt.figure(2)

plt.imshow(img_out)

plt.axis('off')

plt.show()

巴特西

Python: PS 滤镜--旋转模糊

最新文章

热门文章