#-*- coding:utf-8 -*-
import cv2
import numpy as np
import random
def rad(x):
return x*np.pi/180
img = cv2.imread("test0.png")
cv2.imshow("original", img)
#扩展图像，保证内容不超出可视范围
img = cv2.copyMakeBorder(img,100,100,100,100,cv2.BORDER_CONSTANT,(255, 255, 255))
w,h=img.shape[0:2]
cv2.imshow("original1", img)
#anglex=45
#angley = 45
#anglez = 0
#fov = 42
anglex=random.uniform(30,60)
angley =random.uniform(30,60)
anglez =random.uniform(30,60)
fov = random.uniform(30,60)
print(anglex,angley,anglez,fov)
#镜头与图像间的距离，21为半可视角，算z的距离是为了保证在此可视角度下恰好显示整幅图像
z=np.sqrt(w**2 + h**2)/2/np.tan(rad(fov/2))
#齐次变换矩阵
rx = np.array([[1, 0, 0, 0],
[0, np.cos(anglex*np.pi/180), -np.sin(anglex*np.pi/180), 0],
[0, -np.sin(anglex*np.pi/180), np.cos(anglex*np.pi/180), 0,],
[0, 0, 0, 1]], np.float32)
ry = np.array([[np.cos(angley*np.pi/180), 0, np.sin(angley*np.pi/180), 0],
[0, 1, 0, 0],
[-np.sin(angley*np.pi/180),0, np.cos(angley*np.pi/180), 0,],
[0, 0, 0, 1]], np.float32)
rz = np.array([[np.cos(anglez*np.pi/180), np.sin(anglez*np.pi/180), 0, 0],
[-np.sin(anglez*np.pi/180), np.cos(anglez*np.pi/180), 0, 0],
[0, 0, 1, 0],
[0, 0, 0, 1]], np.float32)
r = rx.dot(ry).dot(rz)
#四对点的生成
pcenter = np.array([h/2, w/2, 0, 0], np.float32)
p1 = np.array([0,0, 0,0], np.float32) - pcenter
p2 = np.array([w,0, 0,0], np.float32) - pcenter
p3 = np.array([0,h, 0,0], np.float32) - pcenter
p4 = np.array([w,h, 0,0], np.float32) - pcenter
dst1 = r.dot(p1)
dst2 = r.dot(p2)
dst3 = r.dot(p3)
dst4 = r.dot(p4)
list_dst = [dst1, dst2, dst3, dst4]
org = np.array([[0,0],
[w,0],
[0,h],
[w,h]], np.float32)
dst = np.zeros((4,2), np.float32)
#投影至成像平面
for i in range(4):
dst[i,0] = list_dst[i][0]*z/(z-list_dst[i][2]) + pcenter[0]
dst[i,1] = list_dst[i][1]*z/(z-list_dst[i][2]) + pcenter[1]
warpR = cv2.getPerspectiveTransform(org, dst)
print(org)
print(dst)
result = cv2.warpPerspective(img, warpR, (h,w),(0,255,0))
print(result.shape)
cv2.imshow("result", result)
cv2.imencode(‘.png‘,result)[1].tofile(‘test2.png‘)
c=cv2.waitKey(0)
cv2.destroyAllWindows()
=》 
