8.12.1 cornerHarris.py

# 角点检测:检测图像角点的位置
import cv2
import numpy as np
from matplotlib import pyplot as plt
# 以灰度图模式加载图像
image_bgr = cv2.imread("images/plane_256x256.jpg")
image_gray = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2GRAY)
image_gray = np.float32(image_gray)

# 设置角点检测器的参数
block_size = 2  # 角点检测中窗口的尺寸
aperture = 29   # Sobel算子的尺寸
free_parameter = 0.04   # 用于控制对角点检测的严格程度

# 检测角点
detector_responses = cv2.cornerHarris(image_gray,
                                      block_size,
                                      aperture,
                                      free_parameter)
# 放大角点标志
detector_responses = cv2.dilate(detector_responses,None)
# ----- 显示可能的角点
# plt.imshow(detector_responses, cmap="gray"), plt.axis("off")
# plt.show()


# 只保留大于阈值的检测结果，并把它们标记成白色
threshold = 0.02
image_bgr[detector_responses >
          threshold *
          detector_responses.max()] = [255, 255, 255]
# 转换成灰度图
image_gray = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2GRAY)

plt.imshow(image_gray, cmap="gray"), plt.axis("off")
plt.show()