Selenium Webdriver, numpy 배열로 스크린 샷 (파이썬)

Question

셀레늄 webdriver를 사용하여 스크린 샷을 찍은 다음 저장하지 않고 numpy 배열로 변환하는 방법이 있습니까? openCV와 함께 사용해야합니다.

참고 : 나는 이미지를 저장하고 내가이 일을보다 효율적인 방법이 확신 다시

Answer 1

를 엽니 다 싶지 않다, 그러나 이것은 나를 위해 일한 것입니다 :

from selenium import webdriver 
from PIL import Image 
import StringIO 
import numpy as np 

browser = webdriver.Firefox() 
browser.get('https://www.google.ca/?gws_rd=ssl') 

data = browser.get_screenshot_as_png() 

img = Image.open(StringIO.StringIO(data)) 

numpy_array = np.asarray(img)