### 解决方案 对于在 Linux 的无界面(headless)模式下使用 Selenium 抓取网页图像的问题，可以通过配置 Chrome 或 Firefox 浏览器选项来启用此功能。具体来说，在启动浏览器实例之前设置特定参数允许截图操作即使是在 headless 模式也能正常工作。 #### 配置 Headless 浏览器以支持截屏 为了使无头浏览器能够捕获页面上的视觉内容，需要调整浏览器的初始化过程： - **Chrome 浏览器** 当创建 `webdriver.Chrome` 实例时，应该传递带有适当选项的对象给它。这包括指定 headless 模式以及确保启用了图形渲染能力[^3]。 ```python from selenium import webdriver from selenium.webdriver.chrome.service import Service as ChromeService from selenium.webdriver.chrome.options import Options chrome_options = Options() chrome_options.add_argument("--headless") # 启用无头模式 chrome_options.add_argument('--disable-gpu') # 禁用GPU加速（可选） chrome_options.add_argument("--window-size=1920,1080") # 设置窗口大小 driver = webdriver.Chrome( service=ChromeService(executable_path='./chromedriver'), options=chrome_options ) # 访问目标网站并保存屏幕截图 driver.get('http://example.com') driver.save_screenshot('./screenshot.png') driver.quit() ``` - **Firefox 浏览器** 同样地，如果选择使用 Firefox，则需相应修改其配置项。值得注意的是，在某些情况下还需要额外处理关于显示环境变量的问题[^4]。 ```python from selenium import webdriver from selenium.webdriver.firefox.service import Service as FirefoxService from selenium.webdriver.firefox.options import Options firefox_options = Options() firefox_options.headless = True # 启用无头模式 service = FirefoxService(executable_path='/path/to/geckodriver') driver = webdriver.Firefox(service=service, options=firefox_options) try: driver.set_window_size(1920, 1080) # 可能需要先设定窗口尺寸才能成功截图 driver.get('http://example.com') driver.save_screenshot('/tmp/screenshot_firefox.png') finally: driver.quit() ``` 以上代码片段展示了如何分别针对 Chrome 和 Firefox 设定必要的参数以便于在不依赖 GUI 的环境中执行自动化任务的同时保持获取可视化数据的能力。