在 Jupyter Notebook 中显示 CV2 图像 - Python

在机器学习和计算机视觉的领域中，使用图像处理库 cv2 是常见的。CV2 库可以用于读取、处理和显示图像，但在 Jupyter Notebook 中显示图像需要一些额外的步骤，因为 Jupyter Notebook 显示图像的方式与普通 Python 程序有所不同。

本文将介绍如何在 Jupyter Notebook 中显示 cv2 图像，让您能够更好地了解您的图像处理任务。

如何在 Jupyter Notebook 中显示图像

cv2 库的 imshow() 函数可以用来显示图像，但在 Jupyter Notebook 中必须使用 Matplotlib 库来显示图像。下面的示例代码将演示如何使用 Matplotlib 库在 Jupyter Notebook 中显示 cv2 图像。

import cv2
from matplotlib import pyplot as plt

# read image in grayscale mode
img = cv2.imread('cat.jpg', cv2.IMREAD_GRAYSCALE)

# display image using matplotlib
plt.imshow(img, cmap='gray')
plt.show()

上面的代码中使用了 imread() 函数来读取图像，并将图像转换为灰度模式。imshow() 函数需要使用 gray 颜色图来渲染图像，因此在代码中使用了 cmap='gray'。

使用 plt.show() 函数来在 Jupyter Notebook 中显示图像。

注意事项

1. 如果您在 Jupyter Notebook 中显示彩色图像，则可以在 imshow() 函数中省略 cmap 参数。
2. 在 imshow() 函数中使用灰度颜色图时，您可能会遇到颜色不正确的问题。在这种情况下，您可以使用 cv2.cvtColor() 函数来调整颜色。

import cv2
from matplotlib import pyplot as plt

# read image in color mode
img = cv2.imread('cat.jpg', cv2.IMREAD_COLOR)

# convert image to grayscale using cv2.cvtColor()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# display grayscale image using matplotlib
plt.imshow(gray, cmap='gray')
plt.show()

上面的代码中，我们首先将图像转换为彩色模式，然后使用 cv2.cvtColor() 函数将图像转换为灰度模式。最后，使用 imshow() 函数来显示图像。

结论

在 Jupyter Notebook 中显示 cv2 图像需要使用 Matplotlib 库。使用 imshow() 函数来显示图像，需要注意图像的模式（灰度或彩色）。通过这篇文章，您已经学会了如何在 Jupyter Notebook 中显示 cv2 图像。