Abstract
The target detection algorithm based on deep learning usually has a good effect on traditional target detection, but has a low detection accuracy on small targets. To resolve the problem, this paper proposes an improved YOLOv5 algorithm for small object detection by research on the images captured by UAVs. First, aiming at the problem of high sampling frequency and large image receptive field, upsampling was added to further expand the feature map. Then, to solve the problem of insufficient semantic information of shallow features, feature fusion method was used and one 160*160 output detection layer was added. Finally, since the above steps would increase the amount of computation, the Mobilenet-V2lightweight network was added to improve the detection speed. The VisDrone UAV image data set was used for training and verification and the self-built data set was used for testing. The experiment results show that that improved YOLOv5 algorithm can effectively detect and identify small objects. Compared with traditional detection methods, the small objects detection accuracy and speed is also improved.
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.