偶尔需要用python实现录屏并导出视频,随便写了一个简单的代码。
大概思路是先每隔一段时间截屏,然后将截屏图片变成视频导出。
1 #-*- coding: cp936 -*-
2 from PIL importImageGrab3 importtime,os,cv24 importnumpy as np5 defreleasevideo():6 path=os.getcwd()+'\\picture\\'
7 filelist=os.listdir(path)8 fps=2
9 size=(1366,768)10 video=cv2.VideoWriter("1.avi", cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'), fps, size)11 for item infilelist:12 if item.endswith('.png'):13 #找到路径中所有后缀名为.png的文件,可以更换为.jpg或其它
14 item = path +item15 img =cv2.imread(item)16 video.write(img)17
18 video.release()19 cv2.destroyAllWindows()20 if __name__=='__main__':21 num=022 while 1:23 num+=1
24 time.sleep(0.5)25 bbox=(0,0,1366,768)#四个参数代表了开始截图的x,y,结束截图的x,y,后两个可以看电脑
26 im=ImageGrab.grab(bbox)27 im.save(os.getcwd()+'\\picture\\'+str(num)+'.png')28 if num>20:#这个可以控制时间结束,这边我设置为5秒,改参数就行
29 releasevideo()30 break
31
这里只是测试可行性,没有用优化,事实上没有必要两个分开操作,并行的效率会高很多。利用time.sleep()也是取巧居多。等过段时间利用wxpython写一个python版本的录屏工具。
修正7月8日:
#-*- coding: cp936 -*-
from PIL importImageGrabfrom win32api importGetSystemMetricsimportnumpy as npimportcv2if __name__=='__main__':
num,fps,width,heigh=0,12,GetSystemMetrics(0),GetSystemMetrics(1) #获得分辨率
size=(width,heigh)
video=cv2.VideoWriter("1.avi", cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'), fps, size)while 1:
num+=1bbox=(0,0,width,heigh)#四个参数代表了开始截图的x,y,结束截图的x,y,后两个可以看电脑
im=ImageGrab.grab(bbox)
video.write(np.array(im))#将img convert ndarray
if num>100:#这个可以控制时间结束,日后改成一个按钮
video.release()break
顺带,可以用pip install PyInstall,在.py文件下用pyinstaller -F xxx.py 生成exe文件