Convolutional block attention module based on visual mechanism for robot image edge detection

Abstract

In recent years, with the continuous development of computer vision, digital image and other information technology, its application in robot image has attracted many domestic and foreign scholars to conduct researches. Edge detection technology based on traditional deep learning produces messy and fuzzy edge lines. Therefore, we present a new convolutional block attention module (CBAM) based on visual mechanism for robot image edge detection. CBAM is added into the trunk network, and a down-sampling technique with translation invariance is adopted. Some down-sampling operations in the trunk network are removed to retain the details of the image. Meanwhile, the extended convolution technique is used to increase the model's receptive field. Training is carried out on BSDS500 and PASCAL VOL Context datasets. We use the image pyramid technique to enhance the edges quality during testing. Experimental results show that the proposed model can extract image contour more clearly than other networks, and can solve the problem of edge blur.

Keywords

• deep learning
• CBAM
• visual mechanism
• robot image edge detection