Segmentação semântica de imagens em tempo real baseada em rede de três ramos

REN Fenglei ,  

GAO Ziyang ,  

ZHANG Yan ,  

ZHOU Haibo ,  

YANG Lu ,  

QIN Zhichang ,  

摘要

Devido às exigências rigorosas de precisão e tempo real de algoritmos em cenários de aplicação como percepção ambiental para condução autônoma, foi proposto um algoritmo de segmentação semântica de imagens em tempo real baseado em rede de três ramos para equilibrar eficazmente a precisão do modelo de segmentação semântica e a velocidade de inferência. A estrutura da rede de três ramos é inspirada no algoritmo PIDNet, com ramos destinados respectivamente à extração de detalhes da imagem, informações de contexto semântico e informações de borda. No ramo semântico, foi projetado um módulo eficiente de pooling piramidal para obter informações contextuais em diferentes escalas e ampliar o campo receptivo das características da rede. Nos ramos de detalhes e bordas, foram projetados módulos leves e eficientes de atenção interativa entre canais multiescala para reforçar as características extraídas. Por fim, as características extraídas dos três ramos são fundidas e o resultado final da segmentação semântica é produzido. Os resultados experimentais mostram que o algoritmo proposto baseado na rede de três ramos alcança desempenho de segmentação semântica em tempo real com 79,2% de mIoU e 88,5 quadros/s no conjunto de dados Cityscapes, e 80,5% de mIoU e 140,1 quadros/s no conjunto de dados CamVid. O algoritmo proposto pode executar eficientemente a tarefa de segmentação semântica de imagens, alcançando um excelente equilíbrio entre tempo real e precisão, e superando significativamente os métodos de referência existentes.

关键词

segmentação semântica;aprendizado profundo;tempo real;mecanismo de atenção;características multiescala

阅读全文

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

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