Abstract:Optimizing the structure of deep neural networks is essential in many applications. Especially in the object detection tasks of Unmanned Aerial Vehicles. Due to the constraints of the onboard platform, a more efficient network is required to meet practical demands. Nevertheless, existing lightweight detection networks exhibit excessive redundant computations and may yield in a certain level of accuracy loss. To address these issues, this paper proposes a new lightweight network structure named Cross-Stage Partially Deformable Network (CSPDNet). The initial proposal consists of a Deformable Separable Convolution Block (DSCBlock), separating feature channels, greatly reducing the computational load of convolution, and applying adaptive sampling to the separated feature map. Subsequently, to establish information interaction between feature layers, a channel weighting module is proposed. This module calculates weights for the separated feature map, facilitating information exchange across channels and resolutions. Moreover, it compensates for the effect of point-wise (1 1) convolutions, filtering out more important feature information. Furthermore, a new CSPDBlock is designed, primarily composed of DSCBlock, establishing multidimensional feature correlations for each separated feature layer. This approach improves the ability to capture critical feature information and reconstruct gradient paths, thereby preserving detection accuracy. The proposed technology achieves a balance between model parameter size and detection accuracy. Furthermore, experimental results on object detection datasets demonstrate that our designed network, using fewer parameters, achieves competitive detection performance results compared to existing lightweight networks YOLOv5n, YOLOv6n, YOLOv8n, NanoDet and PP-PicoDet. The optimization effect of the designed CSPDBlock, using the VisDrone dataset, is validated when incorporated into advanced detection algorithms YOLOv5m, PPYOLOEm, YOLOv7, RTMDetm and YOLOv8m. In more detail, by incorporating the designed modules it was achieved that the parameters were reduced by 10–20% while almost maintaining detection accuracy.

LightCSPNet: A Lightweight Network for Image Classification and Objection Detection

LiteCortexNet: Toward Efficient Object Detection at Night

A Lightweight Object Detection Network in Low-Light Conditions Based on Depthwise Separable Pyramid Network and Attention Mechanism on Embedded Platforms.

Small-scale defect detection in industrial environment based on lightweight deep learning network

PCBNet: A Lightweight Convolutional Neural Network for Defect Inspection in Surface Mount Technology

CSPNet: A New Backbone that can Enhance Learning Capability of CNN

LCNet: A Light-Weight Network for Object Counting.

MSPNet: Multi-level Semantic Pyramid Network for Real-Time Object Detection.

ThinNet: an Efficient Convolutional Neural Network for Object Detection

A real-time and efficient surface defect detection method based on YOLOv4

LightNet: A Lightweight 3D Convolutional Neural Network for Real-Time 3D Object Recognition.

LightDefectNet: A Highly Compact Deep Anti-Aliased Attention Condenser Neural Network Architecture for Light Guide Plate Surface Defect Detection

LCNNet: Light-weight Convolutional Neural Networks for Understanding the Highly Congested Scenes.

LARNet:Towards Lightweight, Accurate and Real-time Salient Object Detection

A new lightweight network for efficient UAV object detection

Research on the Construction of an Efficient and Lightweight Online Detection Method for Tiny Surface Defects through Model Compression and Knowledge Distillation

Local and Global Context-Enhanced Lightweight CenterNet for PCB Surface Defect Detection

DLA+: A Light Aggregation Network for Object Classification and Detection

SCFNet: Lightweight Steel Defect Detection Network Based on Spatial Channel Reorganization and Weighted Jump Fusion

Research on Image Classification of Lightweight Convolutional Neural Network

LARNet: Towards Lightweight, Accurate and Real-Time Salient Object Detection