如何用OpenCV提取纯净目标中心线，去除多余分支与噪声？

问题现状 #

现有代码提取鱼的中心线时，存在多余分支与噪声，无法得到纯净的中心线结果。

优化方案 #

针对问题，从预处理、骨架后处理、轮廓筛选三个环节优化：

1. 预处理优化 #

• 替换手动阈值为OTSU自动阈值，适配不同图像的明暗差异
• 使用形态学开运算先去除小噪声点，再用闭运算填补目标区域的微小缺口，减少后续骨架的分支来源

2. 骨架分支修剪 #

利用skimage.morphology.prune_branches对骨架进行修剪，去除短分支，保留主中心线

3. 连通分量筛选 #

提取骨架中的最长连通分量，过滤掉噪声产生的小连通区域

修改后的完整代码 #

import numpy as np
import cv2
from skimage import morphology
from skimage.measure import label, regionprops

# 读取二值化图像
binary_fish = cv2.imread('/content/data/fish12-masked.jpg', cv2.IMREAD_GRAYSCALE)

# 1. 优化预处理
# 形态学开运算：先腐蚀去噪声，再膨胀恢复形状
kernel = np.ones((3, 3), np.uint8)
binary_fish = cv2.morphologyEx(binary_fish, cv2.MORPH_OPEN, kernel, iterations=1)
# OTSU自动阈值二值化
_, binary_fish = cv2.threshold(binary_fish, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)

# 2. 骨架提取+分支修剪
skeleton = morphology.skeletonize(binary_fish > 0)
# 修剪骨架分支，保留主结构
pruned_skeleton = morphology.prune_branches(skeleton)
pruned_skeleton = pruned_skeleton.astype(np.uint8) * 255

# 3. 筛选最长连通分量
# 标记连通区域
labeled_skeleton = label(pruned_skeleton)
# 获取所有连通区域属性
regions = regionprops(labeled_skeleton)
# 筛选出最长的连通区域（按像素数量）
if regions:
    largest_region = max(regions, key=lambda x: x.area)
    # 生成仅包含最长连通分量的图像
    clean_skeleton = np.zeros_like(pruned_skeleton)
    clean_skeleton[labeled_skeleton == largest_region.label] = 255

# 提取中心线轮廓并绘制
contours, _ = cv2.findContours(clean_skeleton, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
centerline = max(contours, key=cv2.contourArea) if contours else []

height, width = binary_fish.shape
centerline_image = np.zeros((height, width), dtype=np.uint8)
if centerline.any():
    cv2.drawContours(centerline_image, [centerline], -1, 255, 1)

# 叠加到原图显示
original_fish = cv2.imread('/content/data/fish12-masked.jpg')
overlay_image = original_fish.copy()
overlay_image[centerline_image > 0] = [0, 0, 255]  # 红色中心线

# 显示结果
cv2.imshow('Clean Centerline', overlay_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

关键优化点说明 #

• OTSU阈值：自动计算最佳二值化阈值，避免手动阈值的局限性
• 开运算：有效去除图像中的小噪声点，减少骨架分支的源头
• 骨架修剪：prune_branches会识别并移除骨架上的短分支，保留主中心线结构
• 最长连通分量筛选：彻底过滤掉噪声产生的孤立小骨架区域

内容的提问来源于stack exchange，提问作者repo_twei