Detecção de alvos em imagens de drones combinando autoatenção e amostragem ramificada

ZHANG Yunzuo ,  

WU Cunyu ,  

LIU Yameng ,  

ZHANG Tian ,  

ZHENG Yuxin ,  

摘要

A detecção de alvos em imagens de drones é amplamente aplicada em muitos campos, mas devido ao complexo fundo da imagem, alta densidade de alvos e variações drásticas de escala, os algoritmos atuais de detecção de alvos em imagens de drones não são suficientemente precisos. Para resolver esses problemas, foi proposta uma metodologia de detecção de alvos em imagens de drones que combina autoatenção e amostragem ramificada. Primeiro, foi projetada uma estrutura residual aninhada que integra autoatenção e convolução para realizar uma combinação eficaz de informações globais e locais, permitindo que o modelo se concentre nos alvos a serem detectados e atenuando o impacto do fundo complexo. Em segundo lugar, foi desenvolvido um módulo de fusão de características baseado em amostragem ramificada para compensar a perda de informações dos alvos. Por fim, foi introduzido um mapa de características granulares superficiais, adicionando uma cabeça de detecção aprimorada para pequenos alvos, a fim de aliviar as mudanças drásticas de escala, e com base nisso, propôs-se um módulo de reforço de características para capturar mais características discriminativas de pequenos alvos. Os experimentos demonstraram que o algoritmo proposto apresenta bom desempenho em diversos cenários. O modelo s alcançou mAP50 e mAP de 59,3% e 37,1% respectivamente no conjunto de dados VisDrone2019, representando um aumento de 5,6% e 5,4% em comparação com o modelo base, e mAP50 e mAP de 44,1% e 24,9% no conjunto de dados UAVDT, melhorando 5,8% e 3,2% em relação ao modelo base.

关键词

imagens de drones;autoatenção;amostragem ramificada;múltiplas escalas;fusão de características

阅读全文

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

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