Object Detection in Security Inspection Scenarios Based on YOLOv5s: Exploring Experiments
Shuoyu Wang, Chenhao Zhang, Zhiren Xiao · Journal of Physics Conference Series · 2023
Abstract At present, there are many problems with the common methods of manual safety checks. Applying object detection technology to X-ray security inspection can effectively improve efficiency and reduce operating costs. The training parameters of deep neural networks directly affect the performance of the network. Based on YOLOv5s network model, this paper uses the method of control variables to explore the impact of training strategies, data enhancement strategies, and loss function selection on network performance. The experimental results show that a suitable batch size and learning rate can achieve significant performance improvements in the model without changing the network structure. With the increase in strength of the data enhancement strategy, the model performance shows a trend of first improving and then decreasing. The selection of the loss function has a great impact on the model’s performance. The research methods and ideas in this article can provide inspiration for subsequent related research.