12 November 2024 Improvement and pruning lightweight research of low-light target detection model based on layer aggregation network and cross-stage partial network
Xinyu Zhao, Pengfei Chang
Author Affiliations +
Abstract

Low-light object detection is critical in computer vision, with widespread applications in nighttime surveillance and autonomous driving. Low-light environments pose several challenges, including increased image noise, insufficient contrast, blurriness, and color distortion, all of which affect target discernibility, particularly for small objects. Furthermore, the requirement for real-time processing adds to the complexity of image handling in these conditions. We introduce a novel approach to improving low-light target detection accuracy while maintaining a compact model. We enhance the YOLOv7-tiny network through several key innovations. We replace the simplified efficient layer aggregation network (ELAN-L) with the enhanced ELAN with fused diverse branch block (ELAN-DBB), which better integrates features from various branches and layers to improve information extraction from low-light images. In addition, we incorporate a new neck structure featuring the efficient cross-stage partial network module integrating Convolutional Block Attention Module and Deformable Convolutional Networks (VoVGSCSP-DCN-CBAM) module. This module combines attention mechanisms and depth-separable convolutions to enhance feature fusion and improve detection performance in low-light conditions. To address spatial awareness, we use CoordConv in place of 1×1 convolutions within the feature pyramid module, which enhances the model’s ability to capture spatial location information crucial for accurate detection. Furthermore, we introduce a specialized pruning module, DBB-pru, which, alongside comparing three pruning methods, results in a 35% reduction in model size and a 3.4% increase in accuracy compared with YOLOv7-tiny. Our approach markedly enhances the precision of low-light target detection while offering a more streamlined and efficacious model for practical deployment in detection applications.

© 2024 SPIE and IS&T
Xinyu Zhao and Pengfei Chang "Improvement and pruning lightweight research of low-light target detection model based on layer aggregation network and cross-stage partial network," Journal of Electronic Imaging 33(6), 063013 (12 November 2024). https://doi.org/10.1117/1.JEI.33.6.063013
Received: 29 July 2024; Accepted: 15 October 2024; Published: 12 November 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
Back to Top