OpenCV库学习之cv2.threshold函数

OpenCV库学习之cv2.threshold函数

一、简介

cv2.threshold是OpenCV库中用于实现图像二值化处理的函数。此函数可以将图像转换为二值图像，即像素值只包含0和最大值（通常是255）的图像。这种转换常用于图像的预处理和特征提取，尤其是在边缘检测和轮廓发现中。

二、语法和参数

ret, dst = cv2.threshold(src, thresh, maxval, type[, dst])

参数:

• src：源图像，必须是单通道灰度图像。
• thresh：阈值，用于确定像素是否应该被视为前景或背景。
• maxval：二值化操作中使用的最大值，通常设为255。
• type：阈值类型，定义了多种二值化方法，包括：
  • cv2.THRESH_BINARY：简单二值化。
  • cv2.THRESH_BINARY_INV：反向二值化。
  • cv2.THRESH_TRUNC：截断二值化，所有高于阈值的像素被设为阈值。
  • cv2.THRESH_TOZERO：高于阈值的像素被设为0。
  • cv2.THRESH_TOZERO_INV：低于阈值的像素被设为0。
  • cv2.THRESH_OTSU：使用Otsu’s方法自动确定阈值。
  • cv2.THRESH_TRIANGLE：使用三角方法自动确定阈值。
• dst：（可选）目标图像，用于存储二值化结果。

返回值：

• ret：返回的阈值，对于自动阈值方法（如cv2.THRESH_OTSU），返回计算得到的阈值。
• dst：二值化后的图像。

三、实例

3.1 简单二值化

import cv2

# 读取灰度图像
image = cv2.imread('path_to_image.jpg', cv2.IMREAD_GRAYSCALE)

# 应用二值化
_, binary_image = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY)

# 显示二值化后的图像
cv2.imshow('Binary Image', binary_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

输出：
图像中像素值高于127的变为255，其余变为0。

3.2 自动阈值二值化（Otsu’s方法）

import cv2

# 读取灰度图像
image = cv2.imread('path_to_image.jpg', cv2.IMREAD_GRAYSCALE)

# 应用Otsu's方法自动二值化
ret, binary_image = cv2.threshold(image, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)

# 显示二值化后的图像
cv2.imshow('Otsu\'s Binary Image', binary_image)
print(f"Automatic Threshold Value: {ret}")
cv2.waitKey(0)
cv2.destroyAllWindows()

输出：
图像自动进行二值化，像素值根据自动计算的阈值进行分类，并打印出该阈值。

四、注意事项

• 确保输入图像是单通道灰度图像，因为cv2.threshold不适用于多通道图像。
• 阈值thresh和最大值maxval应根据实际图像的亮度和对比度进行调整。
• 当使用自动阈值方法时，thresh参数会被忽略，ret将返回计算得到的阈值。
• 在使用cv2.imshow显示图像时，确保使用cv2.waitKey(0)来保持窗口打开直到用户操作。
• 二值化是图像处理中的重要步骤，但可能需要结合其他技术（如滤波、形态学操作）来优化结果。