python怎么实现selenium截图

from pyvirtualdisplay import Display
from selenium import WEBdriver

# 启动虚拟屏幕
display = Display(visible=0, size=(800, 600))
display.start()

# 创建浏览器实例并访问页面
browser = webdriver.Chrome()
browser.get('https://www.Google.com')

# 截图并保存
browser.save_screenshot('screenshot.png')

# 关闭浏览器和虚拟屏幕
browser.quit()
display.stop()

import os
import time
from selenium import webdriver
from pyvirtualdisplay import Display
from PIL import Image

# 设置虚拟屏幕分辨率
display = Display(visible=0, size=(1920, 1080))
display.start()

# 启动浏览器
browser = webdriver.Chrome()

# 打开网页
browser.get('Https://www.baidu.com')

# 等待页面加载完成
time.sleep(5)

# 截图
browser.save_screenshot('screenshot.png')

# 退出浏览器
browser.quit()

# 关闭虚拟屏幕
display.stop()

# 打开截图
Image.open('screenshot.png').show()