CAA-YOLO: Combined-Attention-Augmented YOLO for Infrared Ocean Ships Detection

Abstract

Infrared ocean ships detection still faces great challenges due to the low signal-to-noise ratio and low spatial resolution resulting in a severe lack of texture details for small infrared targets, as well as the distribution of the extremely multiscale ships. In this paper, we propose a CAA-YOLO to alleviate the problems. In this study, to highlight and preserve features of small targets, we apply a high-resolution feature layer (P2) to better use shallow details and the location information. In order to suppress the shallow noise of the P2 layer and further enhance the feature extraction capability, we introduce a TA module into the backbone. Moreover, we design a new feature fusion method to capture the long-range contextual information of small targets and propose a combined attention mechanism to enhance the ability of the feature fusion while suppressing the noise interference caused by the shallow feature layers. We conduct a detailed study of the algorithm based on a marine infrared dataset to verify the effectiveness of our algorithm, in which the AP and AR of small targets increase by 5.63% and 9.01%, respectively, and the mAP increases by 3.4% compared to that of YOLOv5.

Panicle-Cloud: An Open and AI-Powered Cloud Computing Platform for Quantifying Rice Panicles from Drone-Collected Imagery to Enable the Classification of Yield Production in Rice

“…Among the DL models integrated in the Panicle-Cloud platform, the preferred model, Panicle-AI, was developed using the YOLOv5 baseline architecture. Because the successive C3 blocks in the learning architecture could lead to losses of features for small objects [38] , we therefore created a Panicle-Bottleneck (PB) block in each C3 block. The PB block had 2 branches (Fig.…”

A New Vehicle Detection Framework Based on Feature-Guided in the Road Scene

“…Recently, attention mechanisms have gained significant popularity in enhancing the accuracy of object detection in complex backgrounds [25][26][27]. The attention mechanisms enable the weighting of feature information based on learned attention weights.…”

Research on infrared target detection based on neural network fusion

“…Deep learning has become the focus of image processing, it has a strong automatic feature extraction ability [16,17]. The deep learning methods for infrared detection can greatly elevate the detection rate of algorithm [18,19]. In order to further improve the detection rate and enhance the applicability of the detection algorithm, some scholars have proposed fusion network methods for infrared images detection [20][21][22].…”