python - 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)