效果图
原图
仿射后
代码如下
img =cv2.imread("test_03.png")
copy_img = img.copy()
#灰度图
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#高斯滤波
blur = cv2.GaussianBlur(gray , (5,5),0)
#边缘检测
scanny = cv2.Canny(blur,75, 100)
#轮廓检测
ctns = cv2.findContours(scanny.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)[1]
cv2.drawContours(copy_img, ctns, -1, (0,0,255),3)
#排序选出最大外框
ctns = sorted(ctns, key = cv2.contourArea, reverse=True)
#扩张成四个点
if len(ctns)>0:
ctns = sorted(ctns, key = cv2.contourArea, reverse=True)
for c in ctns:
peri = cv2.arcLength(c ,True)
approx = cv2.approxPolyDP(c, .02*peri, True)
if len(approx) ==4:
docCnt = approx
break;
#外框的四个点
(tl,tr,br,bl) = docCnt
#长宽计算，生成模板
widthA = np.sqrt(((br[0] - bl[0]) ** 2) + ((br[1] - bl[1]) ** 2))
widthB = np.sqrt(((tr[0] - tl[0]) ** 2) + ((tr[1] - tl[1]) ** 2))
maxWidth = max(int(widthA), int(widthB))
heightA = np.sqrt(((tr[0] - br[0]) ** 2) + ((tr[1] - br[1]) ** 2))
heightB = np.sqrt(((tl[0] - bl[0]) ** 2) + ((tl[1] - bl[1]) ** 2))
maxHeight = max(int(heightA), int(heightB))
dst = np.asarray([[0,0], [0, maxWidth], [maxHeight, maxWidth],[maxHeight,0]],dtype = "float32")
docCnt = np.float32(docCnt)
#计算变换矩阵
M = cv2.getPerspectiveTransform(docCnt, dst)
#利用变换矩阵计算
wrap = cv2.warpPerspective(gray, M, (maxHeight, maxWidth))
cv2.imshow("wrap",wrap)
cv2.waitKey(10000)
cv2.destroyAllWindows()