Aiming at the problem of large computational resource consumption in the existing object detection network for driver smoking detection, a decomposing YOLOv5 network (Dec-YOLOv5) is proposed for optimization. This method uses singular value decomposition (SVD) to split the standard convolution in the pre-trained YOLOv5 network into two simpler convolution operations to reduce the computational cost. The optimized network generated after decomposition does not need to be retrained, which can reduce the number of parameters and calculations while maintaining the detection accuracy of the pre-trained model. The experimental results show that the detection time of Dec-YOLOv5 is only 80% of the original YOLOv5 when the overall detection accuracy reaches 93.5%. At the same time, compared with the current mainstream object detection model, the Dec-YOLOv5 network has a better ability to express the characteristics of the driver’s cigarettes, and the detection accuracy is higher.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.