本文实例讲述了Python实现PS滤镜Fish lens图像扭曲效果。分享给大家供大家参考，具体如下：

这里实现 PS 滤镜中的一种几何变换– Fish lens， 对图像做扭曲，感觉就像通过一个凸镜或者凹镜在观察图像一样。

import numpy as np

from skimage import img_as_float

import matplotlib.pyplot as plt

from skimage import io

import math

import numpy.matlib

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

img=io.imread(file_name2)

img = img_as_float(img)

row, col, channel = img.shape

img_out = img * 1.0

R=(min(row, col)/2)

# gamma > 1 zoom in gamma < 1 zoom out

gamma = 1.5

center_x = (col-1)/2.0

center_y = (row-1)/2.0

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)

xx_dif = x_mask - center_x

yy_dif = center_y - y_mask

r = np.sqrt(xx_dif * xx_dif + yy_dif * yy_dif)

theta = np.arctan(yy_dif / xx_dif)

mask_1 = xx_dif < 0

theta = theta * (1 - mask_1) + (theta + math.pi) * mask_1

r_new = R*np.power(r/R, gamma)

x_new = r_new * np.cos(theta) + center_x

y_new = center_y - r_new * np.sin(theta)

int_x = np.floor (x_new)

int_x = int_x.astype(int)

int_y = np.floor (y_new)

int_y = int_y.astype(int)

for ii in range(row):

for jj in range (col):

new_xx = int_x [ii, jj]

new_yy = int_y [ii, jj]

if x_new [ii, jj] < 0 or x_new [ii, jj] > col -1 :

continue

if y_new [ii, jj] < 0 or y_new [ii, jj] > row -1 :

continue

img_out[ii, jj, :] = img[new_yy, new_xx, :]

plt.figure (1)

plt.title('www.jb51.net')

plt.imshow (img)

plt.axis('off')

plt.figure (2)

plt.title('www.jb51.net')

plt.imshow (img_out)

plt.axis('off')

plt.show()

运行效果：

希望本文所述对大家Python程序设计有所帮助。