使用OpenCV实现哈哈镜效果

import numpy as np
# Defining the translation matrix# Tx,Ty,Tz represent the position of our virtual camera in the world coordinate systemT = np.array([[1,0,0,-Tx],[0,1,0,-Ty],[0,0,1,-Tz]])
# Defining the rotation matrix # alpha,beta,gamma define the orientation of the virtual camera
Rx = np.array([[1, 0, 0], [0, math.cos(alpha), -math.sin(alpha)], [0, math.sin(alpha), math.cos(alpha)]])
Ry = np.array([[math.cos(beta), 0, -math.sin(beta)],[0, 1, 0],[math.sin(beta),0,math.cos(beta)]])
Rz = np.array([[math.cos(gamma), -math.sin(gamma), 0],[math.sin(gamma),math.cos(gamma), 0],[0, 0, 1]])
R = np.matmul(Rx, np.matmul(Ry, Rz))
# Calculating the extrinsic camera parameter matrix M1M1 = np.matmul(R,T)
# Calculating the intrinsic camera parameter matrix K# sx and sy are apparent pixel length in x and y direction# ox and oy are the coordinates of the optical center in the image plane.K = np.array([[-focus/sx,sh,ox],[0,focus/sy,oy],[0,0,1]])
P = np.matmul(K,RT)

# Determine height and width of input imageH,W = image.shape[:2]# Define x and y coordinate values in range (-W/2 to W/2) and (-H/2 to H/2) respectively
x = np.linspace(-W/2, W/2, W)y = np.linspace(-H/2, H/2, H)
# Creating a mesh grid using the x and y coordinate range defined above.xv,yv = np.meshgrid(x,y)
# Generating the X,Y and Z coordinates of the plane # Here we define Z = 1 planeX = xv.reshape(-1,1)Y = yv.reshape(-1,1)Z = X*0+1 # The mesh will be located on Z = 1 plane
pts3d = np.concatenate(([X],[Y],[Z],[X*0+1]))[:,:,0]
pts2d = np.matmul(P,pts3d)u = pts2d[0,:]/(pts2d[2,:]+0.00000001)v = pts2d[1,:]/(pts2d[2,:]+0.00000001)

pip3 install vcam

import cv2import numpy as npimport mathfrom vcam import vcam,meshGen
# Create a virtual camera object. Here H,W correspond to height and width of the input image frame.c1 = vcam(H=H,W=W)
# Create surface objectplane = meshGen(H,W)
# Change the Z coordinate. By default Z is set to 1# We generate a mirror where for each 3D point, its Z coordinate is defined as Z = 10*sin(2*pi[x/w]*10)plane.Z = 10*np.sin((plane.X/plane.W)*2*np.pi*10)
# Get modified 3D points of the surfacepts3d = plane.getPlane()
# Project the 3D points and get corresponding 2D image coordinates using our virtual camera object c1pts2d = c1.project(pts3d)

# Get mapx and mapy from the 2d projected pointsmap_x,map_y = c1.getMaps(pts2d)
# Applying remap function to input image (img) to generate the funny mirror effectoutput = cv2.remap(img,map_x,map_y,interpolation=cv2.INTER_LINEAR)
cv2.imshow("Funny mirror",output)cv2.waitKey(0)

import cv2import numpy as npimport mathfrom vcam import vcam,meshGen
# Reading the input image. Pass the path of image you would like to use as input image.img = cv2.imread("chess.png")H,W = img.shape[:2]
# Creating the virtual camera objectc1 = vcam(H=H,W=W)
# Creating the surface objectplane = meshGen(H,W)
# We generate a mirror where for each 3D point, its Z coordinate is defined as Z = 20*exp^((x/w)^2 / 2*0.1*sqrt(2*pi))
plane.Z += 20*np.exp(-0.5*((plane.X*1.0/plane.W)/0.1)**2)/(0.1*np.sqrt(2*np.pi))pts3d = plane.getPlane()
pts2d = c1.project(pts3d)map_x,map_y = c1.getMaps(pts2d)
output = cv2.remap(img,map_x,map_y,interpolation=cv2.INTER_LINEAR)
cv2.imshow("Funny Mirror",output)cv2.imshow("Input and output",np.hstack((img,output)))cv2.waitKey(0)

# We generate a mirror where for each 3D point, its Z coordinate is defined as Z = 20*exp^((y/h)^2 / 2*0.1*sqrt(2*pi))
plane.Z += 20*np.exp(-0.5*((plane.Y*1.0/plane.H)/0.1)**2)/(0.1*np.sqrt(2*np.pi))

# We generate a mirror where for each 3D point, its Z coordinate is defined as Z = 20*[ sin(2*pi*(x/w-1/4))) + sin(2*pi*(y/h-1/4))) ]
plane.Z += 20*np.sin(2*np.pi*((plane.X-plane.W/4.0)/plane.W)) + 20*np.sin(2*np.pi*((plane.Y-plane.H/4.0)/plane.H))

# We generate a mirror where for each 3D point, its Z coordinate is defined as Z = -100*sqrt[(x/w)^2 + (y/h)^2]
plane.Z -= 100*np.sqrt((plane.X*1.0/plane.W)**2+(plane.Y*1.0/plane.H)**2)