A lightweight object detection approach based on edge computing for mining industry
Muhammad Wahab Hanif,Zhanli Li,Zhenhua Yu,Rehmat Bashir
DOI: https://doi.org/10.1049/ipr2.13228
IF: 2.3
2024-10-10
IET Image Processing
Abstract:Coal mining enterprises utilize monitoring systems to enhance safety and production efficiency. However, deploying deep learning detection models on edge devices faces significant challenges due to high computational loads, impacting detection speed and accuracy. A mining‐specific dataset, consisting of critical targets like miners, safety helmets, and coal gangue, has been created to overcome these limitations. The proposed model, YOLO‐v4‐LSAM, integrates the Pixel‐wise Normalization Spatial Attention Module (PN‐SAM) into the MobileNet‐v3 bneck structure and replaces the h‐swish activation function with Mish to enhance gradient transfer and focus on key image areas. This model demonstrates a 3.2% improvement in mAP on the VOC2012 dataset and a 2.4% improvement on the mining target dataset, making it highly suitable for real‐time, resource‐constrained edge environments typical of coal mining operations. Coal Mining enterprises deploy numerous monitoring devices to ensure safe and efficient production using target detection technologies. However, deploying deep detection models on edge devices poses challenges due to high computational loads, impacting detection speed and accuracy. A mining target detection dataset has been created to address these issues, featuring key targets in coal mining scenes such as miners, safety helmets, and coal gangue. A model is proposed to improve real‐time performance for edge mining detection tasks. Detection performance is enhanced by incorporating a Pixel‐wise Normalization Spatial Attention Module (PN‐SAM) into the MobileNet‐v3 bneck structure and replacing the h‐swish activation function with Mish, providing more prosperous gradient information transfer. The proposed model, YOLO‐v4‐LSAM, shows a 3.2% mAP improvement on the VOC2012 dataset and a 2.4% improvement on the mining target dataset compared to YOLO‐v4‐Tiny, demonstrating its effectiveness in mining environments. These enhancements enable more accurate and efficient detection in resource‐constrained edge environments, contributing to safer and more reliable monitoring in coal mining operations.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology