[已解决]边缘检测山脊线结果中有很多噪音，怎么消除？请在原有代码基础上给予改正，感谢。

Justheartyoung

代码：
from tifffile import imread, imwrite
from skimage import filters, feature, color
import matplotlib.pyplot as plt
import numpy as np
# 读取TIFF格式无人机影像数据
image = imread('C:\\Users\\WINDOWS\\Desktop\\taiyuan\\python\\1=quick_dom.tif')
# 转换为灰度图像
gray_image = color.rgb2gray(image)
# 边缘检测，提取沙丘脊线
edges = filters.sobel(gray_image)
# 设置合适的阈值来确定沙脊线的二值化图像
threshold = 0.001
binary = edges > threshold
# 叠加沙脊线在原始影像上
overlay = np.copy(image)
overlay[binary] = [255, 0, 0]  # 将沙脊线部分标记为红色
imwrite('edge_001.tif', overlay)
结果：

最佳答案

月排行榜 / 总排行榜

isdkz

2023-8-7 10:48:55

要消除边缘检测结果中的噪音，可以尝试使用图像处理中的腐蚀和膨胀操作。腐蚀可以减小边缘的宽度，同时消除小的噪点，而膨胀可以恢复边缘的宽度，使其更连续。

以下是修改后的代码：

1. from tifffile import imread, imwrite
   
2. from skimage import filters, feature, color, morphology
   
3. import matplotlib.pyplot as plt
   
4. import numpy as np
   
5. 
   
6. # 读取TIFF格式无人机影像数据
   
7. image = imread('C:\\Users\\WINDOWS\\Desktop\\taiyuan\\python\\1=quick_dom.tif')
   
8. 
   
9. # 转换为灰度图像
   
10. gray_image = color.rgb2gray(image)
    
11. 
    
12. # 边缘检测，提取沙丘脊线
    
13. edges = filters.sobel(gray_image)
    
14. 
    
15. # 设置合适的阈值来确定沙脊线的二值化图像
    
16. threshold = 0.001
    
17. binary = edges > threshold
    
18. 
    
19. # 使用腐蚀和膨胀操作消除噪音
    
20. binary = morphology.binary_erosion(binary, morphology.disk(2))  # 腐蚀操作
    
21. binary = morphology.binary_dilation(binary, morphology.disk(2))  # 膨胀操作
    
22. 
    
23. # 叠加沙脊线在原始影像上
    
24. overlay = np.copy(image)
    
25. overlay[binary] = [255, 0, 0]  # 将沙脊线部分标记为红色
    
26. 
    
27. imwrite('edge_001.tif', overlay)
    

复制代码

在代码中，我添加了 morphology.binary_erosion 和 morphology.binary_dilation 两个函数来进行腐蚀和膨胀操作。这里使用了一个半径为2的圆形结构元素，你可以根据实际情况调整结构元素的大小。

通过这样的处理，可以消除边缘检测结果中的噪音，得到更清晰的沙脊线结果。

跳转到最佳答案楼层