工作中遇到了需要使用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这是个什么鬼
