Abstract
The target detection algorithm YOLOv5 is used in real-time object detection with its faster detection speed, good accuracy and flexibility of application. In this paper, a YOLOv5s-SE algorithm is proposed to solve the problem of complex background of X-ray security image and occlusion and overlap of objects. A custom channel attention module is added to the algorithm, which enables the network to focus on feature channels with a large amount of information and suppress unimportant feature channels. The position and quantity of the modules in the network are determined by the method of comparative experiments, and the GIoU loss of target box regression is replaced by CIoU loss to accelerate the convergence of the model. The object detection experiment on the X-ray security image dataset shows that compared with the YOLOv3 and Faster RCNN algorithms, the proposed algorithm has better detection results for partially occluded targets and the case where the distance between the targets is too close. Compared with YOLOv5s, the precision, recall, and mAP are increased by 1.2%, 2.8%, and 2.5% respectively, the loss convergence is also faster.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.