Underwater object detection algorithm based on attention mechanism and cross-stage partial fast spatial pyramidal pooling

The key to precise weeding in the field lies in the efficient detection of weeds. There are no studies on weed detection in potato fields. In view of the difficulties brought by the cross-growth of potatoes and weeds to the detection of weeds, the existing detection methods cannot meet the requirements of detection speed and detection accuracy at the same time. This study proposes an improved YOLOv4 model for weed detection in potato fields. The proposed algorithm replaces the backbone network CSPDarknet53 in the YOLOv4 network structure with the lightweight MobileNetV3 network and introduces Depthwise separable convolutions instead of partial traditional convolutions in the Path Aggregation Network (PANet), which reduces the computational cost of the model and speeds up its detection. In order to improve the detection accuracy, the convolutional block attention module (CBAM) is fused into the PANet structure, and the CBAM will process the input feature map with a channel attention mechanism (CAM) and spatial attention mechanism (SAM), respectively, which can enhance the extraction of useful feature information. The K-means++ clustering algorithm is used instead of the K-means clustering algorithm to update the anchor box information of the model so that the anchor boxes are more suitable for the datasets in this study. Various image processing methods such as CLAHE, MSR, SSR, and gamma are used to increase the robustness of the model, which eliminates the problem of overfitting. CIoU is used as the loss function, and the cosine annealing decay method is used to adjust the learning rate to make the model converge faster. Based on the above-improved methods, we propose the MC-YOLOv4 model. The mAP value of the MC-YOLOv4 model in weed detection in the potato field was 98.52%, which was 3.2%, 4.48%, 2.32%, 0.06%, and 19.86% higher than YOLOv4, YOLOv4-tiny, Faster R-CNN, YOLOv5 l, and SSD(MobilenetV2), respectively, and the average detection time of a single image was 12.49ms. The results show that the optimized method proposed in this paper outperforms other commonly used target detection models in terms of model footprint, detection time consumption, and detection accuracy. This paper can provide a feasible real-time weed identification method for the system of precise weeding in potato fields with limited hardware resources. This model also provides a reference for the efficient detection of weeds in other crop fields and provides theoretical and technical support for the automatic control of weeds.

With the rise of global smart city construction, target detection technology plays a crucial role in optimizing urban functions and improving the quality of life. However, existing target detection technologies still have shortcomings in terms of accuracy, real-time performance, and adaptability. To address this challenge, this study proposes an innovative target detection model. Our model adopts the structure of YOLOv8-DSAF, comprising three key modules: depthwise separable convolution (DSConv), dual-path attention gate module (DPAG), and feature enhancement module (FEM). Firstly, DSConv technology optimizes computational complexity, enabling real-time target detection within limited hardware resources. Secondly, the DPAG module introduces a dual-channel attention mechanism, allowing the model to selectively focus on crucial areas, thereby improving detection accuracy in high-dynamic traffic scenarios. Finally, the FEM module highlights crucial features to prevent their loss, further enhancing detection accuracy. Additionally, we propose an Internet of Things smart city framework consisting of four main layers: the application domain, the Internet of Things infrastructure layer, the edge layer, and the cloud layer. The proposed algorithm utilizes the Internet of Things infrastructure layer, edge layer, and cloud layer to collect and process data in real-time, achieving faster response times. Experimental results on the KITTI V and Cityscapes datasets indicate that our model outperforms the YOLOv8 model. This suggests that in complex urban traffic scenarios, our model exhibits superior performance with higher detection accuracy and adaptability. We believe that this innovative model will significantly propel the development of smart cities and advance target detection technology.

The accuracy and efficiency of object detection is the key technology to spread autonomous driving. In order to address issues such as missed and false detection in traditional target detection algorithms, deep learning technology is used to optimize the ship target detection model as well as the attention mechanisms in this article. Specifically, this paper proposes an improved model based the traditional YOLOv5s algorithm, and a self-created ship dataset is also provided. To refine ship target detection, various attention mechanisms were implemented and compared, including CBAM, ECA, GAM, SimAM, and SK-Net. Through comparative analysis of the effects of these mechanisms, GAM was identified as the optimal choice. The experimental results indicate that the mean average precision (mAP) of detecting ship targets was augmented by 1.0% following the incorporation of GAM. The YOLOv5s-GAM model is therefore deemed as an effective target detection approach in enhancing the safety of ship autopilot, with potential applications in the future.

With the increasing aging of the society, a series of consequences caused by falls of the elderly have become a serious medical problem and a real social problem. Aiming at the fall behavior of the elderly, the research and development of human fall recognition technology has practical application value. This paper proposes a method to recognize human targets in indoor images using target detection technology, and then use the trained neural network based on Self-attention technique named Vision Transformer (ViT) to recognize the falling posture of the elderly. Firstly, the target detection model is optimized and trained for indoor scene, so that the model can accurately detect human targets in indoor scene; Next, the indoor scene human target image is preprocessed, and the indoor human pose image data set is constructed; Then, the data set is used to train the ViT depth network, and enable it to correctly classify five postures (standing, sitting, lying, bending and crawling). The experimental results show that ViT network achieves high accuracy and excellent generalization capacity for the above 5 human posture and can realize the recognition of elderly falls.

With the advent of the big data era, more and more data are used in image detection. In order to solve the problem of small target detection in traditional image detection algorithms, this paper proposes an improved target detection algorithm based on YOLO-v5. The algorithm proposes to connect two models in series, using the former model for target pre-positioning, and the latter model for target detection. This method trains two different model parameters, tests them separately, and finally connects them for target detection. The test results show that the improved algorithm improves the ability in small target detection, and at the same time has a speed advantage.

Highlights MobileNetV3+U-Net lightweight network improves sugarcane segmentation accuracy and speed. The image processing algorithm removes interference information from the predicted sugarcane segmentation images. Improved YOLOX network enhances the accuracy of sugarcane stem node detection and provides better visualization. Hybrid deep learning models are utilized to extract complete feature information required for sugarcane harvesting. Abstract. The efficiency of intelligent sugarcane harvesters in harvesting depends on the effectiveness of identifying and locating the sugarcane during the harvesting process. In the actual harvesting process, accurately extracting valid features of sugarcane amidst the dense and interwoven sugarcane becomes a challenging task. To address this issue, we propose a hybrid deep learning approach to extract sugarcane stem contours and internal stem node feature information from sugarcane efficiently in the context of a complex harvest. Firstly, this study combined the MobileNetV3 and U-Net networks to segment overall images that contain information about the external contours of the sugarcane stem. Then, the extracted overall profile images were optimized using a variety of image processing techniques to meet the requirements of harvesting. Lastly, the improved YOLOX model was utilized to identify the internal stem node features of sugarcane from the optimized overall images. The experimental results on a real sugarcane dataset show that the proposed external sugarcane stem segmentation model achieves a high mean intersection over union (MIoU) of 91.68% with an average segmentation time of just 0.025 seconds. Moreover, the proposed model for internal stem node recognition in sugarcane achieves an average precision (AP) of 96.19% with an average detection time of 0.026 seconds. Additionally, this study compares image segmentation models such as PSPNet and DeepLabv3+ with target detection models such as YoloV5 and YoloV7. The experimental results show that the sugarcane feature extraction models proposed in this article all exhibit high accuracy and robustness.

Chip pad inspection is of great practical importance for chip alignment inspection and correction. It is one of the key technologies for automated chip inspection in semiconductor manufacturing. When applying deep learning methods for chip pad inspection, the main problem to be solved is how to ensure the accuracy of small target pad detection and, at the same time, achieve a lightweight inspection model. The attention mechanism is widely used to improve the accuracy of small target detection by finding the attention region of the network. However, conventional attention mechanisms capture feature information locally, which makes it difficult to effectively improve the detection efficiency of small targets from complex backgrounds in target detection tasks. In this paper, an OCAM (Object Convolution Attention Module) attention module is proposed to build long-range dependencies between channel features and position features by constructing feature contextual relationships to enhance the correlation between features. By adding the OCAM attention module to the feature extraction layer of the YOLOv5 network, the detection performance of chip pads is effectively improved. In addition, a design guideline for the attention layer is proposed in the paper. The attention layer is adjusted by network scaling to avoid network characterization bottlenecks, balance network parameters, and network detection performance, and reduce the hardware device requirements for the improved YOLOv5 network in practical scenarios. Extensive experiments on chip pad datasets, VOC datasets, and COCO datasets show that the approach in this paper is more general and superior to several state-of-the-art methods.

Due to the unique distance and angles involved in satellite remote sensing, ships appear with a small pixel area in images, leading to insufficient feature representation. This results in suboptimal performance in ship detection, including potential misses and false detections. Moreover, the complexity of backgrounds in remote sensing images of ships and the clustering of vessels also adversely affect the accuracy of ship detection. Therefore, this paper proposes an optimized model named SSMA-YOLO, based on YOLOv8n. First, this paper introduces a newly designed SSC2f structure that incorporates spatial and channel convolution (SCConv) and spatial group-wise enhancement (SGE) attention mechanisms. This design reduces spatial and channel redundancies within the neural network, enhancing detection accuracy while simultaneously reducing the model’s parameter count. Second, the newly designed MC2f structure employs the multidimensional collaborative attention (MCA) mechanism to efficiently model spatial and channel features, enhancing recognition efficiency in complex backgrounds. Additionally, the asymptotic feature pyramid network (AFPN) structure was designed for progressively fusing multi-level features from the backbone layers, overcoming challenges posed by multi-scale variations. Experiments of the ships dataset show that the proposed model achieved a 4.4% increase in mAP compared to the state-of-the-art single-stage target detection YOLOv8n model while also reducing the number of parameters by 23%.

Accurate detection of fresh jujube fruits plays a vital role in precision agriculture, enabling reliable yield estimation and supporting automation tasks such as robotic harvesting. To address the challenges of detecting such small targets (≤32 × 32 pixels) in complex orchard environments, this study proposes JFST-DETR, an efficient and robust detection model based on the Real-Time DEtection TRansformer (RT-DETR). First, to address the insufficient feature representation for small jujube fruit targets, a novel module called the Global Awareness Adaptive Module (GAAM) is designed. Building on GAAM and the innovative Spatial Coding Module (SCM), a new Spatial Enhancement Pyramid Network (SEPN) is proposed. Through the spatial-depth transformation domain and global awareness adaptive processing units, SEPN captures fine-grained features of small targets, enhancing the detection accuracy for small objects. Second, a Dynamic Sampling (DySample) operator is adopted, which optimizes feature space details via dynamic offset calculation and lightweight design, improving detection accuracy while reducing computational costs. Finally, to solve the problem of complex background interference caused by foliage occlusion and illumination variations, Pinwheel-Shaped Convolution (PSConv) is introduced. By using asymmetric padding and multi-directional convolution, PSConv enhances the robustness of feature extraction, ensuring reliable recognition in complex agricultural environments. Experimental results show that JFST-DETR achieves precision, recall, F1, mAP@50, and mAP@50:95 of 93%, 86.8%, 89.8%, 94.3%, and 75.2%. Compared to the baseline model, these metrics improve by 0.8%, 3.7%, 2.4%, 2.6%, and 3.1%, respectively. Cross-dataset evaluations further confirm its strong generalizability, demonstrating potential as a practical solution for small-target detection in intelligent horticulture.

Accurate bud detection is a prerequisite for automatic tea picking and yield statistics; however, current research suffers from missed detection due to the variety of singleness and false detection under complex backgrounds. Traditional target detection models are mainly based on CNN, but CNN can only achieve the extraction of local feature information, which is a lack of advantages for the accurate identification of targets in complex environments, and Transformer can be a good solution to the problem. Therefore, based on a multi-variety tea bud dataset, this study proposes RT-DETR-Tea, an improved object detection model under the real-time detection Transformer (RT-DETR) framework. This model uses cascaded group attention to replace the multi-head self-attention (MHSA) mechanism in the attention-based intra-scale feature interaction (AIFI) module, effectively optimizing deep features and enriching the semantic information of features. The original cross-scale feature-fusion module (CCFM) mechanism is improved to establish the gather-and-distribute-Tea (GD-Tea) mechanism for multi-level feature fusion, which can effectively fuse low-level and high-level semantic information and large and small tea bud features in natural environments. The submodule of DilatedReparamBlock in UniRepLKNet was employed to improve RepC3 to achieve an efficient fusion of tea bud feature information and ensure the accuracy of the detection head. Ablation experiments show that the precision and mean average precision of the proposed RT-DETR-Tea model are 96.1% and 79.7%, respectively, which are increased by 5.2% and 2.4% compared to those of the original model, indicating the model’s effectiveness. The model also shows good detection performance on the newly constructed tea bud dataset. Compared with other detection algorithms, the improved RT-DETR-Tea model demonstrates superior tea bud detection performance, providing effective technical support for smart tea garden management and production.

With the rapid development of UAV and infrared imaging technology, the cost of UAV infrared imaging technology has decreased steadily. Small target detection technology in aerial infrared images has great potential for applications in many fields, especially in the field of traffic surveillance. Because of the low contrast and relatively limited feature information in infrared images compared to visible images, the difficulty involved in small road target detection in infrared aerial images has increased. To solve this problem, this study proposes a feature-enhanced attention and dual-GELAN net (FEADG-net) model. In this network model, the reliability and effectiveness of small target feature extraction is enhanced by a backbone network combined with low-frequency enhancement and a swin transformer. The multi-scale features of the target are fused using a dual-GELAN neck structure, and a detection head with the parameters of the auto-adjusted InnerIoU is constructed to improve the detection accuracy for small infrared targets. The viability of the method was proved using the HIT-UAV dataset and IRTS-AG dataset. According to a comparative experiment, the mAP50 of FEADG-net reached more than 90 percent, which was higher than that of any previous method and it met the real-time requirements. Finally, an ablation experiment was conducted to demonstrate that all three of the modules proposed in the method contributed to the improvement in the detection accuracy. This study not only designs a new algorithm for small road object detection in infrared remote sensing images from UAVs but also provides new ideas for small target detection in remote sensing images for other fields.

Sonar is the primary tool used for detecting small targets at long distances underwater. Due to the influence of the underwater environment and imaging mechanisms, sonar images face challenges such as a small number of target pixels, insufficient data samples, and uneven category distribution. Existing target detection methods are unable to effectively extract features from sonar images, leading to high false positive rates and affecting the accuracy of target detection models. To counter these challenges, this paper presents a novel sonar small-target detection framework named DCE-Net that refines the YOLOv8 architecture. The Detail Enhancement Attention Block (DEAB) utilizes multi-scale residual structures and channel attention mechanism (AM) to achieve image defogging and small-target structure completion. The lightweight spatial variation convolution module (CoordGate) reduces false detections in complex backgrounds through dynamic position-aware convolution kernels. The improved efficient multi-scale AM (MH-EMA) performs scale-adaptive feature reweighting and combines cross-dimensional interaction strategies to enhance pixel-level feature representation. Experiments on a self-built sonar small-target detection dataset show that DCE-Net achieves an mAP@0.5 of 87.3% and an mAP@0.5:0.95 of 41.6%, representing improvements of 5.5% and 7.7%, respectively, over the baseline YOLOv8. This demonstrates that DCE-Net provides an efficient solution for underwater detection tasks.

It is difficult to identify the ship images obtained by a synthetic aperture radar (SAR) due to the influence of dense ships, complex background and small target size, so a deep learning-based target detection algorithm was introduced to obtain better detection performance. However, in order to achieve excellent performance, most of the current target detection algorithms focus on building deep and high-width neural networks, resulting in bloated network structure and reduced detection speed, which is not conducive to the practical application of target detection algorithms. Thereby, an efficient lightweight network Efficient-YOLO for ship detection in complex situations is proposed in the present work. Firstly, a new regression loss function ECIOU is proposed to enhance the detection boxes localization accuracy and model convergence speed. Secondly, We propose the SCUPA module to enhance the multiplexing of picture feature information and the model generalization performance. Thirdly, The GCHE module is proposed to strengthen the network’s ability to extract feature information. At last, the effectiveness of our method is tested on the specialized ship dataset: SSDD and HRSID datasets. The results show that Efficient-YOLO outperforms other state-of-the-art algorithms in accuracy, recall and detection speed, with smaller model complexity and model size.

The task of object detection is to accurately and efficiently identify and locate a large number of predefined categories of object instances from images. With the wide application of deep learning, the accuracy and efficiency of target detection have been greatly improved. However, deep learning-based target detection still faces challenges from key technologies such as improving and optimizing the performance of mainstream target detection algorithms, improving the detection accuracy of small target objects, realizing multi-class object detection and lightweight detection model. In response to the above challenges, based on extensive literature research, this paper analyses methods for lightweight detection models and improved detection accuracy from the perspective of YOLOv5s network structure. The problems to be solved in target detection and the future research direction are predicted and prospected.

An improved YOLOv8s-based method is proposed to address the challenge of accurately recognizing tiny objects in remote sensing images during practical human-computer interaction. In detecting tiny targets, the accuracy of YOLOv8s is low because the downsampling module of the original YOLOv8s algorithm causes the network to lose fine-grained feature information, and the neck network feature information needs to be sufficiently fused. In this method, the strided convolution module in YOLOv8s is replaced with the SPD-Conv module. By doing so, the feature map undergoes downsampling while preserving fine-grained feature information, thereby improving the learning and expressive capabilities of the network and enhancing recognition accuracy. Meanwhile, the path aggregation network is substituted with the SPANet structure, which facilitates the acquisition of more prosperous gradient paths. This substitution enhances the fusion of feature maps at various scales, reduces model parameters, and further improves detection accuracy. Additionally, it enhances the network’s robustness to complex backgrounds. Experimental verification is conducted on the following two intricate datasets containing tiny objects: AI-TOD and TinyPerson. A comparative analysis with the original YOLOv8s algorithm reveals notable enhancements in recognition accuracy. Specifically, under real-time performance constraints, the proposed method yields a 4.9% and 9.1% improvement in mAP0.5 recognition accuracy for AI-TOD and TinyPerson datasets, respectively. Moreover, the recognition accuracy for mAP0.5:0.95 is enhanced by 3.4% and 3.2% for the same datasets, respectively. The results indicate that the proposed method enables rapid and accurate recognition of tiny objects in complex backgrounds. Furthermore, it demonstrates better recognition precision and stability than other algorithms, such as YOLOv5s and YOLOv8s.