python 仿射变换 伪彩上色

通过仿射变换将扭曲的图片拉正，并填上伪彩色，对应四个点的坐标可以通过imshow打开图片后将鼠标放置在相应点上来记录，也可以通过电脑中的画图软件打开然后找点。

代码：

import cv2
import numpy as np

img = cv2.imread('3.jpg')
#img2 = cv2.imread('2.jpg')
'''cv2.namedWindow('img',0)
cv2.namedWindow('img2',0)
cv2.imshow('img',img)
cv2.imshow('img2',img2)'''

height1 = img.shape[0]
width1 = img.shape[1]
ImgCopy = img.copy()
#gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
src = np.float32([[0,0],[321,100],[100,213],[640,417]])
dst = np.float32([[0,0],[640,0],[0,417],[640,417]])
m = cv2.getPerspectiveTransform(src,dst)


img3 = cv2.warpPerspective(ImgCopy,m,(width1,height1))
gray = cv2.cvtColor(img3,cv2.COLOR_BGR2GRAY)
img_color = cv2.applyColorMap(gray,cv2.COLORMAP_TWILIGHT)

cv2.imshow('img',img)
cv2.imshow('img3',img3)
cv2.imshow('img4',img_color)
#cv2.imshow('img4',img4)
cv2.waitKey()
cv2.destroyAllWindows()