opencv python在视屏上截图

介绍

因为我是一个小白，所以代码就比较简单，如果有错误，希望大家可以评论下告诉我，感激。

效果

蓝色框框是我们要截下的图

这是截完后的图

代码

import cv2
import numpy as np
import matplotlib.pyplot as plt
i = w = h = f1 = f2 = ew = eh = 0

def cut(event, x, y, flags, param):
    global i,w,h, f1,f2,ew,eh
    img = 0
    #这一个if的作用是，当鼠标左键按下，记录开始的位置
    if event == cv2.EVENT_LBUTTONDOWN:
        f1 = 1
        h = x
        w = y
        print("w = ", w, "h = ", h)
    #这一步的作用是当鼠标滑动时，并且左键是按下的时候，就画一个框框，并且记录左后一个位置
    if event == cv2.EVENT_MOUSEMOVE and flags == cv2.EVENT_FLAG_LBUTTON:
        img = frame.copy()
        cv2.rectangle(img, (h,w), (x,y), (255,0,0),5)
        cv2.imshow("frame", img)
        f2 = 1
        eh = x
        ew = y
    #当执行了以上的两步时候，并且左键是松开的，就可以进行截出一张图片。
    if f1 == 1 and f2 == 1 and event == cv2.EVENT_LBUTTONUP:
        #下面两个if是向各个方向上截图的考虑
        if ew < w:
            w,ew = ew,w
        if eh < h:
            eh,h = h, eh
        i += 1
        f1 = f2 = 0
        img = frame[w:ew, h:eh]
        path = "C:/test2/" + "%d"%i+".jpg"
        cv2.imwrite(path,img)
#这时创建一个对像
cap = cv2.VideoCapture("C:/test2/bbb.mp4")
#这是命名一个窗口，与下面的鼠标结合，鼠标会在这个窗口中产生效果
cv2.namedWindow("frame",cv2.WINDOW_NORMAL)
cv2.setMouseCallback("frame",cut)
while True:
	#读取图片
    ret, frame = cap.read()
    if ret == False:
        break
    cv2.imshow("frame", frame)
    #如果输入'c'，就进行切割。
    if cv2.waitKey(10) == ord("c"):
        cv2.waitKey(0)
    if cv2.waitKey(10) == ord('q'):
        break

注意

x,y和 w,h, 的不同，x 代表的图像的列，y是代表着行
frame[w:ew, h:eh]这个先是行，再是列。