OpenCV-Python实战|Shi-Tomasi拐角探测器:用goodFeaturesToTrack解锁更精准的特征点提取

pythoncv2.goodFeaturesToTrack(image, maxCorners, qualityLevel, minDistance, corners=None, mask=None, blockSize=None, useHarrisDetector=None, k=None)

python# 安装/升级OpenCV-Python（核心依赖）pip install opencv-python -U# 安装辅助库（图像显示、数据处理）pip install numpy matplotlib

pythonimport cv2import numpy as npimport matplotlib.pyplot as plt# 1. 读取图像并预处理（灰度化+降噪，衔接前文预处理知识点）img = cv2.imread("chessboard.jpg") # 替换为自己的图像路径（棋盘/建筑均可）gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 必须转为单通道灰度图# 高斯平滑降噪（减少噪声干扰，提升检测精度，与哈里斯实操一致）gray_blur = cv2.GaussianBlur(gray, (3, 3), sigmaX=1)# 2. 执行Shi-Tomasi拐角检测（核心步骤，调用goodFeaturesToTrack()）maxCorners = 50 # 最多保留50个优质角点qualityLevel = 0.01 # 角点质量阈值，0.01~0.1为宜minDistance = 10 # 两个角点最小距离，避免冗余blockSize = 3 # 检测窗口大小，默认3# 调用函数，提取优质角点corners = cv2.goodFeaturesToTrack(image=gray_blur,maxCorners=maxCorners,qualityLevel=qualityLevel,minDistance=minDistance,blockSize=blockSize,useHarrisDetector=False # 默认为False，启用Shi-Tomasi算法)# 3. 处理角点坐标（转换为整数类型，便于绘制）corners = np.int0(corners) # 转换为int0类型（适配OpenCV绘制函数）# 4. 标记角点（蓝色实心圆，清晰可见，便于观察）for corner in corners:x, y = corner.ravel() # 提取角点坐标（x, y）# 绘制实心圆：参数（图像，坐标，半径，颜色，厚度-1表示实心）cv2.circle(img, (x, y), 3, (255, 0, 0), -1)# 5. 显示效果（原始图 vs 角点检测结果）plt.figure(figsize=(12, 6))# 原始图像plt.subplot(1, 2, 1)plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))plt.title("原始图像（棋盘）")plt.axis("off")# Shi-Tomasi检测结果plt.subplot(1, 2, 2)plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))plt.title("Shi-Tomasi拐角检测结果（蓝色实心圆）")plt.axis("off")plt.tight_layout()plt.show()# 可选：保存检测结果cv2.imwrite("shi_tomasi_basic_result.jpg", img)

pythonimport cv2import numpy as npimport matplotlib.pyplot as plt# 读取不同场景的测试图（覆盖3种典型场景）img1 = cv2.imread("chessboard.jpg") # 场景1：角点密集、清晰（棋盘）img2 = cv2.imread("building.jpg") # 场景2：角点稀疏、轮廓清晰（建筑）img3 = cv2.imread("natural.jpg") # 场景3：角点模糊、噪声较多（自然场景）# 定义分场景调优参数（核心3个参数）scenes = [# (图像, 场景名称, maxCorners, qualityLevel, minDistance)(img1, "棋盘场景", 80, 0.01, 8),(img2, "建筑场景", 30, 0.02, 15),(img3, "自然场景", 20, 0.03, 20)]# 批量处理不同场景，执行Shi-Tomasi检测plt.figure(figsize=(15, 5))for i, (img, name, maxCorners, qualityLevel, minDistance) in enumerate(scenes):gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)gray_blur = cv2.GaussianBlur(gray, (3, 3), sigmaX=1)# 调用函数，使用调优后的参数corners = cv2.goodFeaturesToTrack(gray_blur, maxCorners=maxCorners, qualityLevel=qualityLevel,minDistance=minDistance, blockSize=3)corners = np.int0(corners)# 标记角点for corner in corners:x, y = corner.ravel()cv2.circle(img, (x, y), 3, (255, 0, 0), -1)# 显示效果plt.subplot(1, 3, i+1)plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))plt.title(f"{name}（调优后）")plt.axis("off")plt.tight_layout()plt.show()

pythonimport cv2import numpy as npimport matplotlib.pyplot as plt# 1. 读取图像并预处理img_harris = cv2.imread("chessboard.jpg") # 哈里斯检测图像img_shi = cv2.imread("chessboard.jpg") # Shi-Tomasi检测图像gray = cv2.cvtColor(img_harris, cv2.COLOR_BGR2GRAY)gray_blur = cv2.GaussianBlur(gray, (3, 3), sigmaX=1)# 2. Shi-Tomasi检测（useHarrisDetector=False，默认）corners_shi = cv2.goodFeaturesToTrack(gray_blur, maxCorners=50, qualityLevel=0.01, minDistance=10, blockSize=3)corners_shi = np.int0(corners_shi)for corner in corners_shi:x, y = corner.ravel()cv2.circle(img_shi, (x, y), 3, (255, 0, 0), -1) # 蓝色：Shi-Tomasi# 3. 哈里斯检测（useHarrisDetector=True）corners_harris = cv2.goodFeaturesToTrack(gray_blur, maxCorners=50, qualityLevel=0.01, minDistance=10,blockSize=3, useHarrisDetector=True, k=0.04)corners_harris = np.int0(corners_harris)for corner in corners_harris:x, y = corner.ravel()cv2.circle(img_harris, (x, y), 3, (0, 0, 255), -1) # 红色：哈里斯# 4. 显示对比效果plt.figure(figsize=(14, 6))# Shi-Tomasi检测结果plt.subplot(1, 2, 1)plt.imshow(cv2.cvtColor(img_shi, cv2.COLOR_BGR2RGB))plt.title("Shi-Tomasi拐角检测（蓝色，精准无冗余）")plt.axis("off")# 哈里斯检测结果plt.subplot(1, 2, 2)plt.imshow(cv2.cvtColor(img_harris, cv2.COLOR_BGR2RGB))plt.title("哈里斯角检测（红色，少量冗余）")plt.axis("off")plt.tight_layout()plt.show()# 可选：保存对比结果cv2.imwrite("shi_tomasi_vs_harris.jpg", np.hstack((img_shi, img_harris)))

pythonimport cv2import numpy as npimport matplotlib.pyplot as plt# 1. 环境验证（可选）print("OpenCV版本：", cv2.__version__)# 2. 读取图像（替换为自己的图像路径）img_basic = cv2.imread("chessboard.jpg") # 基础实操图像img_scene1 = cv2.imread("chessboard.jpg") # 场景1：棋盘img_scene2 = cv2.imread("building.jpg") # 场景2：建筑img_scene3 = cv2.imread("natural.jpg") # 场景3：自然img_compare1 = cv2.imread("chessboard.jpg")# 对比图：Shi-Tomasiimg_compare2 = cv2.imread("chessboard.jpg")# 对比图：哈里斯if None in [img_basic, img_scene1, img_scene2, img_scene3, img_compare1, img_compare2]:print("错误：未找到图像，请检查图像路径！")else:# 3. 基础Shi-Tomasi检测gray_basic = cv2.cvtColor(img_basic, cv2.COLOR_BGR2GRAY)gray_basic_blur = cv2.GaussianBlur(gray_basic, (3, 3), sigmaX=1)corners_basic = cv2.goodFeaturesToTrack(gray_basic_blur, maxCorners=50, qualityLevel=0.01, minDistance=10)corners_basic = np.int0(corners_basic)for corner in corners_basic:x, y = corner.ravel()cv2.circle(img_basic, (x, y), 3, (255, 0, 0), -1)# 4. 分场景调优检测scenes = [(img_scene1, 80, 0.01, 8),(img_scene2, 30, 0.02, 15),(img_scene3, 20, 0.03, 20)]scene_results = []for img, maxCorners, qualityLevel, minDistance in scenes:gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)gray_blur = cv2.GaussianBlur(gray, (3, 3), sigmaX=1)corners = cv2.goodFeaturesToTrack(gray_blur, maxCorners=maxCorners, qualityLevel=qualityLevel,minDistance=minDistance, blockSize=3)corners = np.int0(corners)for corner in corners:x, y = corner.ravel()cv2.circle(img, (x, y), 3, (255, 0, 0), -1)scene_results.append(img)# 5. Shi-Tomasi与哈里斯对比gray_compare = cv2.cvtColor(img_compare1, cv2.COLOR_BGR2GRAY)gray_compare_blur = cv2.GaussianBlur(gray_compare, (3, 3), sigmaX=1)# Shi-Tomasicorners_shi = cv2.goodFeaturesToTrack(gray_compare_blur, maxCorners=50, qualityLevel=0.01, minDistance=10)corners_shi = np.int0(corners_shi)for corner in corners_shi:x, y = corner.ravel()cv2.circle(img_compare1, (x, y), 3, (255, 0, 0), -1)# 哈里斯corners_harris = cv2.goodFeaturesToTrack(gray_compare_blur, maxCorners=50, qualityLevel=0.01, minDistance=10,useHarrisDetector=True, k=0.04)corners_harris = np.int0(corners_harris)for corner in corners_harris:x, y = corner.ravel()cv2.circle(img_compare2, (x, y), 3, (0, 0, 255), -1)# 6. 统一显示所有效果plt.figure(figsize=(18, 12))# 第一行：基础检测+对比plt.subplot(3, 4, 1)plt.imshow(cv2.cvtColor(img_basic, cv2.COLOR_BGR2RGB))plt.title("基础Shi-Tomasi检测")plt.axis("off")plt.subplot(3, 4, 2)plt.imshow(cv2.cvtColor(img_compare1, cv2.COLOR_BGR2RGB))plt.title("Shi-Tomasi（蓝色）")plt.axis("off")plt.subplot(3, 4, 3)plt.imshow(cv2.cvtColor(img_compare2, cv2.COLOR_BGR2RGB))plt.title("哈里斯（红色）")plt.axis("off")# 第二行：分场景调优plt.subplot(3, 4, 5)plt.imshow(cv2.cvtColor(scene_results[0], cv2.COLOR_BGR2RGB))plt.title("棋盘场景（maxCorners=80）")plt.axis("off")plt.subplot(3, 4, 6)plt.imshow(cv2.cvtColor(scene_results[1], cv2.COLOR_BGR2RGB))plt.title("建筑场景（maxCorners=30）")plt.axis("off")plt.subplot(3, 4, 7)plt.imshow(cv2.cvtColor(scene_results[2], cv2.COLOR_BGR2RGB))plt.title("自然场景（maxCorners=20）")plt.axis("off")plt.tight_layout()plt.show()# 7. 保存结果（可选）cv2.imwrite("shi_tomasi_basic.jpg", img_basic)cv2.imwrite("shi_tomasi_vs_harris.jpg", np.hstack((img_compare1, img_compare2)))cv2.imwrite("shi_tomasi_scene1.jpg", scene_results[0])cv2.imwrite("shi_tomasi_scene2.jpg", scene_results[1])cv2.imwrite("shi_tomasi_scene3.jpg", scene_results[2])