OpenCV 录制视屏随记

工作中遇到了需要使用opencv来录制屏幕的功能,所以记录下来, 防止下次还能遇到忘记了。。。。。我这脑子肯定会忘

# -*- coding: utf-8 -*-
# -*- coding: utf-8 -*-
from PIL import ImageGrab
import numpy as np
import cv2
import os
import time

filepath = 'C:\\Users\Administrator\\'

class ScreenRecord():
    """ For recording what happened in your screen """
    def __init__(self):
        pass
    
    def record(self):
        self.recordStop = False
        screen = ImageGrab.grab()
        k = np.zeros((1, 1), np.uint8)
        length, width = screen.size
        video_decode_style = cv2.VideoWriter_fourcc('D', 'I', 'V', 'X')

        time_now = time.strftime("%Y-%m-%d-%H_%M_%S",time.localtime(time.time())) 
        video_name = filepath + time_now + r'.mp4'
        if os.path.exists(video_name):
            os.remove(video_name)
        video = cv2.VideoWriter(video_name, video_decode_style, 1, (length, width))
        while self.recordStop == False:
            im = ImageGrab.grab()
            imm = cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)
            video.write(imm)
            cv2.imshow('imm', k)
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break

        video.release()
        cv2.destroyAllWindows()

    def start(self):
        self.record()

    def stop(self):
        try:
            os._exit(0)
        except:
            print ('die.')

if __name__ == '__main__':
    ScreenRecord().start()
  • 最重要的是VideoWriter_fourcc()这个设置,当对视屏的大小有需求的时候,就需要关注视屏的编码问题,opencv感觉上对于主流的解码方式有支持但是编码过后的视屏还是存在过大的现象
  • imshow()会出现一个弹窗来显示出正在录制的状况,对于这个窗口的出现还需要继续研究一下,opencv-python的官方文档上的使用就是这样的,我就拿过来用了一下。
  • 这个脚本最麻烦的地方就是怎么样使视频更小,编码方式,帧率都会影响。我把帧率调到了最低。以确保视频最小
  • 当使用pyinstaller打包成exe文件的时候,需要将文件的后缀改为.mp4。TMD这是个什么鬼
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容