### 使用Python实现ANMS算法进行特征点检测的经典应用 #### 输入图片的ANMS实现 下面是一个完整的流程，展示如何使用Python实现自适应非极大值抑制（ANMS）算法来进行特征点检测。此过程包括读取输入图像、提取Harris角点或其他特征点，并通过ANMS筛选出最优子集。 #### 完整代码示例 ```python import cv2 import numpy as np from matplotlib import pyplot as plt from scipy.spatial import cKDTree def compute_harris_response(gray_img, sigma=1.0): """ 计算 Harris 角点响应图。 参数: gray_img: 单通道灰度图像 (numpy array) sigma: 高斯核的标准差 返回: harris_response: Harris 响应矩阵 """ # 导数计算 grad_x = cv2.Sobel(gray_img, cv2.CV_64F, 1, 0, ksize=3) grad_y = cv2.Sobel(gray_img, cv2.CV_64F, 0, 1, ksize=3) Ixx = grad_x ** 2 Ixy = grad_x * grad_y Iyy = grad_y ** 2 # 高斯平滑 gauss_Ixx = cv2.GaussianBlur(Ixx, (5, 5), sigma) gauss_Ixy = cv2.GaussianBlur(Ixy, (5, 5), sigma) gauss_Iyy = cv2.GaussianBlur(Iyy, (5, 5), sigma) det_M = gauss_Ixx * gauss_Iyy - gauss_Ixy ** 2 trace_M = gauss_Ixx + gauss_Iyy harris_response = det_M - 0.04 * (trace_M ** 2) return harris_response def detect_corners(harris_response, threshold=0.01): """ 检测超过阈值的角点。 参数: harris_response: Harris 响应矩阵 threshold: 响应值的相对阈值 返回: corners: 初步检测到的角点列表 [(y, x, response_value)] """ max_response = np.max(harris_response) corner_mask = harris_response > threshold * max_response corner_coords = np.argwhere(corner_mask).astype(np.float32) responses = harris_response[corner_mask].flatten() corners = list(zip(corner_coords[:, 0], corner_coords[:, 1], responses)) return corners def anms(corners, num_best=500): """ 自适应非极大值抑制 (ANMS)，选择最均匀分布的角点。 参数: corners: 初始角点列表 [(y, x, response_value)] num_best: 输出的最佳角点数量上限 返回: best_corners: 筛选后的最佳角点列表 [(y, x, response_value)] """ points = np.array([[c[1], c[0]] for c in corners]) scores = np.array([c[2] for c in corners]) tree = cKDTree(points) min_distances = [] for i, p in enumerate(points): distances, indices = tree.query(p, k=len(points)) r_i = None for d, idx in zip(distances, indices): if scores[i] <= scores[idx]: r_i = d break if r_i is not None: min_distances.append((i, r_i)) sorted_min_distances = sorted(min_distances, key=lambda x: -x[1])[:num_best] best_indices = [idx for idx, _ in sorted_min_distances] best_corners = [corners[i] for i in best_indices] return best_corners # 主程序逻辑 if __name__ == "__main__": # 加载图像并转换为灰度模式 image_path = 'input_image.jpg' # 替换为实际路径 img = cv2.imread(image_path) gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY).astype(np.float32) # Step 1: 计算 Harris 角点响应 harris_response = compute_harris_response(gray_img) # Step 2: 检测初步角点 initial_corners = detect_corners(harris_response) # Step 3: 应用 ANMS 进一步筛选 final_corners = anms(initial_corners, num_best=200) # 显示结果 fig, ax = plt.subplots(1, figsize=(10, 10)) ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)) ax.scatter( [c[1] for c in final_corners], [c[0] for c in final_corners], color='red', s=10, label="Selected Corners" ) ax.legend() plt.show() ``` --- ### 关键说明 上述代码实现了以下主要功能模块： 1. **Harris角点检测**：通过梯度信息和结构张量计算得到角点响应值[^1]。 2. **初步角点过滤**：设定阈值挑选出潜在候选角点。 3. **ANMS优化**：采用空间距离约束机制剔除冗余点，保留分布更均匀的一组特征点[^2]。 整个过程中涉及到了OpenCV库用于基础图像处理操作以及Matplotlib用来可视化最终的结果。 --- ###