Abstract:In the operation site of power grid construction, it is crucial to comprehensively and efficiently detect violations of regulations for the personal safety of the workers with a safety monitoring system based on object detection technology. However, common general-purpose object detection algorithms are difficult to deploy on low-computational-power embedded platforms situated at the edge due to their high model complexity. These algorithms suffer from drawbacks such as low operational efficiency, slow detection speed, and high energy consumption. To address this issue, a lightweight violation detection algorithm based on the SP (Segmentation-and-Product) attention mechanism, named SP-YOLO-Lite, is proposed to improve the YOLOv5s detection algorithm and achieve low-cost deployment and efficient operation of object detection algorithms on low-computational-power monitoring platforms. First, to address the issue of excessive complexity in backbone networks built with conventional convolutional modules, a Lightweight Convolutional Block was employed to construct the backbone network, significantly reducing computational and parameter costs while maintaining high detection model accuracy. Second, in response to the problem of existing attention mechanisms overlooking spatial local information, we introduced an image segmentation operation and proposed a novel attention mechanism called Segmentation-and-Product (SP) attention. It enables the model to effectively capture local informative features of the image, thereby enhancing model accuracy. Furthermore, a Neck network that is both lightweight and feature-rich is proposed by introducing Depthwise Separable Convolution and Segmentation-and-Product attention module to Path Aggregation Network, thus addressing the issue of high computation and parameter volume in the Neck network of YOLOv5s. Experimental results show that compared with the baseline network YOLOv5s, the proposed SP-YOLO-Lite model reduces the computation and parameter volume by approximately 70%, achieving similar detection accuracy on both the VOC dataset and our self-built SMPC dataset.

An improved insulator self-explosion detection method based on group-level pruning for the YOLOv7-tiny algorithm

Lightweight detection method for industrial gas leakage based on improved YOLOv7-tiny

Vehicle and Pedestrian Detection Based on Improved YOLOv7-Tiny

An improved lightweight object detection algorithm for YOLOv5

Re-Parameterization After Pruning: Lightweight Algorithm Based on UAV Remote Sensing Target Detection

Lightweight target detection algorithm based on YOLOv4

A lightweight fire detection algorithm for small targets based on YOLOv5s

Lightweight Vehicle Detection Based on Improved YOLOv5s

FL-YOLOv7: A Lightweight Small Object Detection Algorithm in Forest Fire Detection

Dense-YOLOv7: improved real-time insulator detection framework based on YOLOv7

Group channel pruning and spatial attention distilling for object detection

Lightweight Transmission Line Fault Detection Method Based on Leaner YOLOv7-Tiny

Infra-YOLO: Efficient Neural Network Structure with Model Compression for Real-Time Infrared Small Object Detection

Lightweight coal and gangue detection algorithm based on improved Yolov7-tiny

SP-YOLO-Lite: A Lightweight Violation Detection Algorithm Based on SP Attention Mechanism

An Improved Lightweight YOLOv8 Network for Early Small Flame Target Detection

Insulator and Defect Detection Model Based on Improved Yolo-S

A Lightweight Vehicle Detection Method Fusing GSConv and Coordinate Attention Mechanism

The improved YOLOv8 algorithm based on EMSPConv and SPE-head modules

Fire Detection Method Based on Lightweight YOLOv4

Real-time object detection method based on YOLOv5 and efficient mobile network