一、柱形畸变

import cv2
import numpy as np

def cylindrical_projection(image, f):
    h, w = image.shape[:2]
    map_x, map_y = np.meshgrid(np.arange(w), np.arange(h))
    
    x_c = w / 2
    y_c = h / 2
    
    theta = (map_x - x_c) / f
    h_c = (map_y - y_c) / np.sqrt((map_x - x_c)**2 + f**2)
    
    map_x_cyl = f * np.tan(theta) + x_c
    map_y_cyl = f * h_c + y_c
    
    return cv2.remap(image, map_x_cyl.astype(np.float32), map_y_cyl.astype(np.float32), cv2.INTER_LINEAR)


# 读取图像
image = cv2.imread(r"/media/face/Face/Face/ABERDEEN_Face/ABERDEEN/adrian1.jpg")

# 设置焦距
focal_length = 120

# 进行柱面投影
result = cylindrical_projection(image, focal_length)

# 显示结果
cv2.imwrite("test.jpg",result)

def cylindrical_projection(image, f):
    h, w = image.shape[:2]
    map_x, map_y = np.meshgrid(np.arange(w), np.arange(h))
    
    x_c = w / 2
    y_c = h * 1.5
    
    theta = (map_x - x_c) / f
    h_c = (map_y - y_c) / np.sqrt((map_x - x_c)**2 + f**2)
    
    map_x_cyl = f * np.tan(theta) + x_c
    map_y_cyl = f * h_c + y_c