Algorithme léger pour la détection de petites cibles dans les images de drones

LUO Kexin ,  

LI Songjiang ,  

WANG Peng ,  

YANG Huamin ,  

摘要

Pour résoudre le problème de détection insuffisante des cibles, notamment des petites cibles, dans les images de drones avec les modèles de détection existants, cet article propose un modèle léger de détection de petites cibles PRSU-YOLO. Tout d'abord, un module convolutif adaptatif en forme d'éolienne a été conçu pour améliorer la capacité d'extraction des caractéristiques fines ; ensuite, un module convolutionnel d'espace-canal reparamétré a été construit pour optimiser la représentation multi-échelle par reconstruction dynamique des caractéristiques ; de plus, une branche renforcée de détection des petites cibles a été intégrée dans le réseau du cou pour construire un chemin de renforcement des caractéristiques détaillées à haute résolution ; enfin, une fonction de perte dynamique de rapport d'intersection sur union a été introduite pour permettre au modèle d'ajuster de manière adaptative la stratégie de régression des boîtes englobantes. Sur la base d'une complexité de calcul n’augmentant que de 14,6 GFLOPS, notre modèle atteint un mAP@0.5 de 37,4% sur le jeu de données VisDrone2019, soit une amélioration de 4,4% par rapport au modèle de référence ; la précision sur le jeu TinyPerson atteint 34,5%, une augmentation de 3,5%. Les résultats expérimentaux montrent que ce modèle améliore significativement les capacités de détection tout en contrôlant efficacement les coûts de calcul, fournissant une solution efficace pour les tâches de détection d’objets dans les scénarios d’observation terrestre par drones.

关键词

vision par ordinateur;images de drones;détection de petites cibles;renforcement des caractéristiques

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website