1. 程式人生 > 程式設計 >python 實現波浪濾鏡特效

python 實現波浪濾鏡特效

本文用 Python 實現 PS 濾鏡的波浪特效

import numpy as np
from skimage import img_as_float
import matplotlib.pyplot as plt
from skimage import io
import numpy.matlib
import math
# 圖片位置
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
alpha = 70.0
beta = 30.0
degree = 20.0

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,1)
y_mask = np.transpose(y_mask)

xx_dif = x_mask - center_x
yy_dif = center_y - y_mask

x = degree * np.sin(2 * math.pi * yy_dif / alpha) + xx_dif
y = degree * np.cos(2 * math.pi * xx_dif / beta) + yy_dif

x_new = x + center_x
y_new = center_y - y 

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.imshow (img)
plt.axis('off')

plt.figure (2)
plt.imshow (img_out)
plt.axis('off')

plt.show()

實現效果

python 實現波浪濾鏡特效

以上就是python 實現波浪濾鏡特效的詳細內容,更多關於python 波浪濾鏡的資料請關注我們其它相關文章!