视频流 实现四边形透视变换标准矩阵(python-opencv)

功能 :实现视频流中的四边形–》》标准矩阵变换

一、简单介绍

要点 :1、鼠标回调函数获取四边形四点坐标值(list型)
2、将四边形四点坐标值(list型)转换为array型
3、通过透视函数变换实现四边形-》矩阵过程

使用 :1、鼠标双击四个坐标点,依次为左上、右上、左下、右下
(1)----------------(2)**
(1)-------------------(2)
|----------------------|
(3)-------------------(4)
2、再次双击任意位置确认定标完成

二、实现效果图

(用杰伦做效果展示嘻嘻嘻)
1、第一帧定标图
视频流 实现四边形透视变换标准矩阵(python-opencv)_第1张图片
2、视频流定标
视频流 实现四边形透视变换标准矩阵(python-opencv)_第2张图片
3、视频流矩阵变换输出效果
视频流 实现四边形透视变换标准矩阵(python-opencv)_第3张图片

三、代码实现

import cv2
import numpy as np

camera = cv2.VideoCapture(1)     #摄像头
# camera = cv2.VideoCapture('E:/视频/quanjing.mp4')    #视频流
count = 0       #鼠标事件计数值(左键双击一次计数值加一)
Recpoints = []  #空列表用于储存原始图像四点坐标
point2 = np.array([[0, 0], [320, 0], [0, 180], [320, 180]], dtype="float32")

def draw_circle(event,x,y,flags,param):
    global count
    global Recpoints
    if event==cv2.EVENT_LBUTTONDBLCLK:
        count = count + 1
        # print('jishu',count)
        # print('[',x,',',y,']')
        Recpoints.append([x,y])
        print(Recpoints)
        cv2.circle(img,(x,y),10,(0,0,255),-1)

ret,img = camera.read()   #img作为第一帧选定坐标点的图像
cv2.namedWindow('image',cv2.WINDOW_NORMAL)
cv2.namedWindow('video',cv2.WINDOW_NORMAL)
cv2.setMouseCallback('image',draw_circle)

ret, frame = camera.read()
while(count < 5):
    cv2.imshow('image',img)
    point1 = np.array(Recpoints,dtype = "float32")
    for (x, y) in Recpoints:
        cv2.circle(frame, (x, y), 10, (0, 255, 0), -1)
    cv2.imshow('video',frame)
    ret, frame = camera.read()

    if cv2.waitKey(20)&0xFF==27:  #按ESC退出
        break

print('dingbiao-end')   #四个坐标点定标结束
# print('pit1',point1)
# print('pit2',point2)
cv2.destroyWindow('image')    #关闭前两个窗口
cv2.destroyWindow('video')

cv2.namedWindow('out', cv2.WINDOW_NORMAL)  #透视变换后窗口

while(1):
    #透视变换
    ret, frame = camera.read()
    w = frame.shape[0]
    h = frame.shape[1]
    M = cv2.getPerspectiveTransform(point1, point2)
    out_img = cv2.warpPerspective(frame, M, (w, h))
    cv2.imshow('out', out_img)
    ret, image = camera.read()

    if cv2.waitKey(20) & 0xFF == 27:  # 按ESC退出
        break
cv2.destroyAllWindows()
``
#  以上代码仅供参考


你可能感兴趣的:(图像处理,Python学习,图像变换,opencv,python,计算机视觉)