使用Python保存数组中的屏幕截图

时间:2017-07-27 02:59:53

标签: python arrays screenshot cv2 python-mss

如何使用python,mss和opencv捕获我的电脑屏幕并将其保存为一组图像以形成电影?我正在转换为灰度级,因此它可以是一个三维数组。我想将每个2d屏幕截图存储在3d数组中以供查看和处理。我很难构建一个保存屏幕截图序列的数组,以及在cv2中播放屏幕截图的顺序。 非常感谢

import time
import numpy as np
import cv2
import mss
from PIL import Image

with mss.mss() as sct:
    fps_list=[]
    matrix_list = []
    monitor = {'top':40, 'left':0, 'width':800, 'height':640}
    timer = 0
    while timer <100:
        last_time = time.time()

        #get raw pizels from screen and save to numpy array
        img = np.array(sct.grab(monitor))
        img=cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        #Save img data as matrix
        matrix_list[timer,:,:] = img

        #Display Image
        cv2.imshow('Normal', img)
        fps = 1/ (time.time()-last_time)
        fps_list.append(fps)

        #press q to quit
        timer += 1
        if cv2.waitKey(25) & 0xFF == ord('q'):
            cv2.destroyAllWindows()
            break
#calculate fps
fps_list = np.asarray(fps_list)
print(np.average(fps_list))


#playback image movie from screencapture
t=0
while t < 100:
    cv.imshow('Playback',img_matrix[t])
    t += 1

2 个答案:

答案 0 :(得分:0)

使用collections.OrderedDict()保存序列

import collections
....
fps_list= collections.OrderedDict()
...
fps_list[timer] = fps

答案 1 :(得分:0)

也许是一个线索,将截图保存到列表中并稍后重播(您将不得不调整睡眠时间):

import time
import cv2
import mss
import numpy


with mss.mss() as sct:
    monitor = {'top': 40, 'left': 0, 'width': 800, 'height': 640}
    img_matrix = []

    for _ in range(100):
        # Get raw pizels from screen and save to numpy array
        img = numpy.array(sct.grab(monitor))
        img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        # Save img data as matrix
        img_matrix.append(img)

        # Display Image
        cv2.imshow('Normal', img)

        # Press q to quit
        if cv2.waitKey(25) & 0xFF == ord('q'):
            cv2.destroyAllWindows()
            break

    # Playback image movie from screencapture
    for img in img_matrix:
        cv2.imshow('Playback', img)
        time.sleep(0.1)