红外技术, 2019, 41 (1): 72, 网络出版: 2019-03-23
一种改进的 HED网络及其在边缘检测中的应用
Research on Significant Edge Detection of Infrared Image
Based on Deep Learning
深度学习 HED网络 红外图像 显著性边缘 边缘检测 deep learning holistically nested edge detection network infrared image saliency edge edge detection
摘要
HED网络( holistically nested edge detection network)被证明是目前用于边缘检测的一种性能较好的深度学习网络, 但在实际应用中发现, 将该网络用于前下视红外成像制导自动目标识别时, 会出现检测出的边缘不完整、不光滑等问题。针对上述问题, 对 HED网络进行了改进, 在此基础上提出了一种基于改进 HED网络的边缘提取方法。首先, 在原网络结构的基础上减少了两个池化层, 提高了侧边输出层的输出精度, 然后, 将改进 HED网络输出的边缘概率图进行二值化, 得到显著性边缘区域; 最后, 采用基于匹配滤波的边缘提取方法提取图像的边缘, 并将其与改进 HED网络提取的显著目标边缘相融合, 得到最终结果。实验结果表明, 该方法能够大幅减少非目标区域的边缘, 并且能够有效提取较为完整和准确的目标边缘, 为后续红外图像中的目标检测、跟踪与识别奠定了良好的基础。
Abstract
The holistically nested edge detection (HED) network has proven to be a better deep-learning network for edge detection so far. However, it is found that the detected edges are incomplete and not smooth when the network is used for automatic target recognition in forward-looking infrared imaging guidance. To solve these problems, the HED network needs to be improved. Hence, an edge extraction method based on the improved HED network is proposed. First, two pooling layers are reduced on the basis of the original HED network structure, and the output accuracy of the side output layer is improved. Then, the edge probability map of the output of the improved HED network is binarized to obtain a significant edge region. Finally, the edge extracted by a matching filter is merged with the edge extracted by the improved HED network. The experimental results show that this method can significantly reduce the edges of non-target areas, and can effectively extract relatively complete and accurate target edges, laying a good foundation for target detection, tracking, and recognition in subsequent infrared images.
焦安波, 何淼, 罗海波. 一种改进的 HED网络及其在边缘检测中的应用[J]. 红外技术, 2019, 41(1): 72. JIAO Anbo, HE Miao, LUO Haibo. Research on Significant Edge Detection of Infrared Image Based on Deep Learning[J]. Infrared Technology, 2019, 41(1): 72.