在Google Colab运行图像去雾代码时cv2.imshow相关报错问题

问题根源 #

image_dehazer库的remove_haze函数内部硬编码调用了cv2.imshow，而Colab环境不支持该函数（会抛出cv2.error: OpenCV(...) can't open display类错误）。手动修改库文件后仍报错，大概率是因为你修改的不是Colab当前运行环境加载的库文件副本，或是修改后未重启运行时生效。

可行解决方案 #

方案1：动态替换cv2.imshow（无需修改源文件） #

在调用去雾函数前，临时将cv2.imshow替换为cv2_imshow，执行完成后可恢复原函数，无需改动库源码：

from google.colab.patches import cv2_imshow
import cv2
import image_dehazer

# 保存原生cv2.imshow
original_imshow = cv2.imshow
# 替换为Colab兼容的cv2_imshow
cv2.imshow = cv2_imshow

# 执行去雾操作
HazeCorrectedImg, haze_map = image_dehazer.remove_haze(input_image)

# 可选：恢复原生cv2.imshow（若后续有需求）
cv2.imshow = original_imshow

方案2：修改Colab环境中加载的库文件 #

如果方案1无效，直接定位并修改Colab当前环境的库文件：

1. 先打印image_dehazer库的安装路径：

import image_dehazer
print(image_dehazer.__file__)

2. 复制输出的路径（例如/usr/local/lib/python3.10/dist-packages/image_dehazer/__init__.py），用Colab左侧文件面板的编辑器打开该文件，找到所有cv2.imshow的调用行，替换为cv2_imshow（需确保已导入from google.colab.patches import cv2_imshow）。
3. 修改保存后，重启Colab运行时（菜单栏Runtime -> Restart runtime），再重新执行你的代码。

方案3：安装自定义修改版库 #

若不想每次实例重置后重复操作，可forkimage_dehazer源码仓库，替换所有cv2.imshow为cv2_imshow，然后在Colab中安装修改后的版本：

!pip uninstall -y image_dehazer
!pip install git+https://github.com/你的用户名/你的image_dehazer仓库.git

注意事项 #

• Colab的运行环境是临时的，方案2的修改在实例重置后会失效，需重新操作。
• 方案1是最便捷的临时解决方式，适合单次运行场景。

内容的提问来源于stack exchange，提问作者Z hades