本文实例讲述了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.****.net/matrix_space/article/details/72301893