有趣的photoshop特效（1）：凸透镜python实现

一、概念

ps中有凸透镜效果，PS菜单栏->虑镜->扭曲->球面化，也称为鱼眼效果，即通过凸透镜所看到的视觉效果。下面来用算法来实现该效果。

二、算法思想

主要应用的是图像插值算法，以图像中的一个点为中心，该中心即为透镜中心，慢慢的往四周插值扩散，如下所示。

实现需要一个映射函数，将原坐标映射到外围的对应坐标，参考凸透镜效果算法matlab版。很多函数，只要是凸函数（函数的二阶导数大于等于零），都可以实现以上功能，比如y=x^2/r； y=x^3/r； y = r*log(x+1) / log (r+1)，这里我用的是y = x^2/r。其中x是原图像坐标到圆心的距离，y是映射后的新坐标到圆心的距离。

三、效果

 

四、Python代码

import cv2
import numpy as np
img = cv2.imread('D:/Code/python/Data/dog.jpg')
#img = cv2.imread('me.jpg')
row, col, dim = img.shape
new_img = np.zeros([row, col, dim], dtype=np.uint8)

#cv2.imshow('img', img)
#cv2.waitKey(0)

#
center_i = row / 2
center_j = col / 2
radius = min(center_i, center_j) # 防止前面赋值越界

# i, j为new_img的坐标，下面要找到原图像的对应坐标
for i in range(row):
    for j in range(col):
        y = np.sqrt((i - center_i)**2 + (j - center_j)**2) # 新坐标i,j与圆心距离
        new_img[i, j, :] = img[i, j, :]
        if (y <= radius): # 圆心内的像素要分配新的位置
            original_i = np.int(np.floor(y*(i-center_i) / radius + center_i)) # 根据现有坐标找回原图坐标，这里不懂，记录下来
            original_j = np.int(np.floor(y*(j-center_j) / radius + center_j))
            new_img[i, j, :] = img[original_i, original_j, :]

cv2.imshow('new_img', new_img)
cv2.waitKey(0)