图片转换特效 水波 波浪滤镜

[python]代码库

#!/usr/bin/env python
# encoding: utf-8
'''
图片转换特效 水波特效滤镜
'''
import cv2, math
import numpy as np


class picture:

    def __init__(self):
        self.path = 'assets/picture.jpeg'

    def hello(self):
        return self

    def run(self):
        # 读取原始图像
        img = cv2.imread(self.path)
        # 获取图像行和列
        rows, cols = img.shape[:2]
        # 新建目标图像
        dst = np.zeros((rows, cols, 3), dtype="uint8")
        # 定义水波特效参数
        #越大越疏
        wavelength = 50
        #越大越密
        amplitude = 30
        phase = math.pi / 1.1
        # 获取中心点
        centreX = 0.5
        centreY = 0.5
        radius = min(rows, cols) / 2
        # 设置水波覆盖面积
        icentreX = cols * centreX
        icentreY = rows * centreY

        # 图像水波特效
        for i in range(rows):
            for j in range(cols):
                dx = j - icentreX
                dy = i - icentreY
                distance = dx * dx + dy * dy

                if distance > radius * radius:
                    x = j
                    y = i
                else:
                    # 计算水波区域
                    distance = math.sqrt(distance)
                    amount = amplitude * math.sin(distance / wavelength * 2 * math.pi - phase)
                    amount = amount * (radius - distance) / radius
                    amount = amount * wavelength / (distance + 0.00001)
                    x = j + dx * amount
                    y = i + dy * amount
                # 边界判断
                if x < 0:
                    x = 0
                if x >= cols - 1:
                    x = cols - 2
                if y < 0:
                    y = 0
                if y >= rows - 1:
                    y = rows - 2
                p = x - int(x)
                q = y - int(y)

                # 图像水波赋值
                dst[i, j, :] = (1 - p) * (1 - q) * img[int(y), int(x), :] + p * (1 - q) * img[int(y), int(x), :]
                + (1 - p) * q * img[int(y), int(x), :] + p * q * img[int(y), int(x), :]

        # 显示图像
        cv2.imshow('src', img)
        cv2.imshow('dst', dst)
        cv2.waitKey()
        cv2.destroyAllWindows()


if __name__ == '__main__':
    picture().hello().run()

[代码运行效果截图]