Método de detección de objetos YOLOv3 con módulo de autoatención con máscara gaussiana

KONG Ya-jie ,  

ZHANG Ye ,  

摘要

El método de detección de objetos basado en imágenes de conducción proporciona una solución económica y eficaz para percibir el entorno vial circundante, pero también impone altos requisitos en la efectividad y velocidad de la detección. Este artículo estudia el algoritmo de detección de objetos de una etapa basado en aprendizaje profundo YOLOv3, combinando un mecanismo de autoatención. Se insertó un módulo de atención automática con máscara gaussiana en las capas profundas de su red para mitigar la limitación del campo receptivo de las operaciones convolucionales, capturando más información global y mejorando el rendimiento del algoritmo. Los resultados experimentales muestran que el modelo mejorado alcanzó un mAP@0.5 de 56.88% y una precisión del 65.31% en el conjunto de datos MS COCO 2017. En comparación con YOLOv3, el mAP@0.5 aumentó un 2.56% y la precisión un 3.53%. Aunque la velocidad de detección sufrió una leve pérdida, se mejoró eficazmente el rendimiento del modelo, lo que puede soportar mejor aplicaciones como la conducción asistida.

关键词

Detección de objetos;Red YOLOv3;Autoatención;Conducción asistida

阅读全文

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

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