OpenCV中的置换贴图过滤器

import cv2
import numpy as np

# 读取原始图像
image = cv2.imread('input.jpg')

# 创建一个高度图（灰度图像）
height_map = np.zeros_like(image)
height_map[:, :] = np.random.randint(0, 255, size=image.shape[:2]).astype(np.uint8)

# 将高度图转换为浮点数类型
height_map = height_map.astype(np.float32) / 255.0

# 计算位移量
displacement_x = cv2.Sobel(height_map, cv2.CV_32F, 1, 0)
displacement_y = cv2.Sobel(height_map, cv2.CV_32F, 0, 1)

# 应用位移
rows, cols, _ = image.shape
map_x = np.zeros((rows, cols), np.float32)
map_y = np.zeros((rows, cols), np.float32)
for i in range(rows):
    for j in range(cols):
        map_x[i, j] = j + displacement_x[i, j] * 10
        map_y[i, j] = i + displacement_y[i, j] * 10

# 使用remap函数进行图像变形
result = cv2.remap(image, map_x, map_y, cv2.INTER_LINEAR)

# 显示结果
cv2.imshow('Result', result)
cv2.waitKey(0)
cv2.destroyAllWindows()