计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (6): 219-224.DOI: 10.3778/j.issn.1002-8331.1912-0162

• 工程与应用 • 上一篇    下一篇

基于改进显著图的高效裂纹检测算法

王翀,韩振奇,徐浩煜,祝永新,徐胜,陈夏   

  1. 1.中国科学院 上海高等研究院,上海 201210
    2.中国科学院大学,北京 100049
    3.联想集团研究院 上海分院,上海 201210
    4.上海微小卫星工程中心,上海 201210
  • 出版日期:2021-03-15 发布日期:2021-03-12

Efficient Crack Detection Algorithm Based on Improved Saliency Map

WANG Chong, HAN Zhenqi, XU Haoyu, ZHU Yongxin, XU Sheng, CHEN Xia   

  1. 1.Shanghai Advanced Research Institute, Chinese Academy of Sciences, Shanghai 201210, China
    2.University of Chinese Academy of Sciences, Beijing 100049, China
    3.Shanghai Branch, Lenovo Research, Shanghai 201210, China
    4.Microsatellites of Chinese Academy of Sciences, Shanghai 201210, China
  • Online:2021-03-15 Published:2021-03-12

摘要:

裂纹检测对于确保设施和设备的安全运行非常重要。然而现有的裂纹检测方法数据标注困难,且训练时间长,检测速度慢。针对这些问题,提出一种基于改进显著图的裂纹检测算法,应用卷积神经网络提取的特征计算出图像中裂纹的位置,从而实现裂纹检测。该算法解决了裂纹检测算法数据标注困难的问题;训练时间短,使用两块k80显卡训练时间仅为23 min;检测速度快,具有105 帧/s的检测速度;同时在自制数据集上达到了52%的平均精度,验证了该算法的有效性。

关键词: 裂纹检测, 卷积神经网络, 显著图, 像素级检测, 实时检测

Abstract:

Crack detection is significant important for the safe operation of facilities and equipment. However, some problems exist in current crack detection methods:the data annotation process is cumbersome; the training process is time-consuming; and the detection speed is slow. In order to solve these problems, a weakly supervised crack detection algorithm is proposed. The algorithm uses the features extracted by the convolutional neural network to generate a saliency map containing the position information of the crack in the image, thus achieving crack detection. The experimental results show that the crack detection algorithm based on improved saliency map solves the difficulty in data annotation, and has a pixel-level detection effect while only need to learn image categories. The training time is short, only needs 23 minutes using two k80 GPUs, at the same time, achieving 52% average precision on the homemade data set and has a detection speed of 105 frames per second, which verifies the effectiveness of the algorithm.

Key words: crack detection, convolutional neural network, saliency map, pixel-level detection, real-time detection