本文实例讲述了Python实现PS滤镜Fish lens图像扭曲效果。分享给大家供大家参考,具体如下:
这里实现 PS 滤镜中的一种几何变换– Fish lens, 对图像做扭曲,感觉就像通过一个凸镜或者凹镜在观察图像一样。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
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.zyiz.net' ) plt.imshow (img) plt.axis( 'off' ) plt.figure ( 2 ) plt.title( 'www.zyiz.net' ) plt.imshow (img_out) plt.axis( 'off' ) plt.show() |
运行效果:
希望本文所述对大家Python程序设计有所帮助。
原文链接:http://blog.csdn.net/matrix_space/article/details/72301893