Smart City Gnosys
Smart city article details
| Title | Small Target Detection Algorithm Based On Improved Yolov5 |
|---|---|
| ID_Doc | 49082 |
| Authors | Li Y.; Zhu Y.; Hu P.; Wang H.; Ding H. |
| Year | 2024 |
| Published | Proceedings of SPIE - The International Society for Optical Engineering, 13224 |
| DOI | http://dx.doi.org/10.1117/12.3034943 |
| Abstract | At present, the Internet and smart cities are developing rapidly, and all social services. Small target detection is widely used in these fields. This paper focuses on small target detection based on YOLOv5. The feature enhancement module is proposed to solve the problem of incomplete extraction of small target features by the model. An attention mechanism is added to the model, so that the detection model pays more attention to the region in the image that contains the target to be detected, increases the weight of the feature information in this part, enriches the feature space, and further improves the performance of small target detection. Finally, by combining the use of network pruning and knowledge distillation, the model is compressed to compress the model size and improve the detection speed under the premise of ensuring that the model detection accuracy is not affected. The experimental results show that our optimisation effectively improves small target detection in terms of both detection accuracy and detection speed, with FPS improved by 31.84 and AP improved by 1.7%. © 2024 SPIE. |
| Author Keywords | attention mechanism; feature enhancement; model lightweighting; Small target detection |
Similar Articles
| Id | Similarity | Authors | Title | Published |
|---|---|---|---|---|
38160  | 0.856 | Wang Y.; Liu X.; Wang R. | Multi-Class Object Detection In Urban Scenes Based On Deep Learning | ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences, 10, 3 (2024) |
| 5341 | 0.854 | Mei L.; Zhang J.; Long T. | A Survey On The Development And Application Of Yolov5 In The Internet Of Things | Proceedings - 2024 International Conference on Identification, Information and Knowledge in the Internet of Things, IIKI 2024 (2024) |
| 62125 | 0.854 | Song F.; Li P. | Yolov5-Ms: Real-Time Multi-Surveillance Pedestrian Target Detection Model For Smart Cities | Biomimetics, 8, 6 (2023) |
| 25460 | 0.851 | Zhao Q.; Ma W.; Zheng C.; Li L. | Exploration Of Vehicle Target Detection Method Based On Lightweight Yolov5 Fusion Background Modeling | Applied Sciences (Switzerland), 13, 7 (2023) |
| 45926 | 0.85 | Guo R.; Li S.; Wang K. | Research On Yolov3 Algorithm Based On Darknet Framework | Journal of Physics: Conference Series, 1629, 1 (2020) |