Selenium Webdriver，截图为numpy数组（Python）

Question

有没有办法用selenium 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)