Rede de detecção de nuvem de pontos Pointpillars baseada em destilação de conhecimento e orientação de localização

ZHAO Jing ,  

LI Shaobo ,  

GUO Jielong ,  

YU Hui ,  

ZHANG Jianfeng ,  

LI Jie ,  

摘要

Os dados LiDAR, devido às suas características geométricas, são amplamente utilizados em tarefas de detecção de objetos tridimensionais. Devido à escassez e irregularidade dos dados de nuvens de pontos, é difícil alcançar um equilíbrio entre a qualidade da extração de características e a velocidade de inferência. Este artigo propõe um algoritmo de detecção de objetos 3D baseado em codificação de características de pilares volumétricos, baseado na rede Pointpillars, com um framework de modelo professor-aluno para a destilação da escala da caixa de regressão, aumentando a perda de destilação, otimizando o treinamento do modelo de rede e melhorando a qualidade da extração de características. Para melhorar ainda mais o desempenho da detecção do modelo, foi projetado um termo de classificação guiado por localização, aumentando a correlação entre a previsão de classificação e a de regressão, elevando a precisão do reconhecimento de objetos. As melhorias feitas nesta rede não introduzem embeddings de rede adicionais. Os resultados experimentais no conjunto de dados KITTI mostram que, no modo 3D, o valor médio de precisão aumentou de 60,65% para 64,69%, e no modo visão aérea, o valor médio de precisão aumentou de 67,74% para 70,24%, em comparação com a rede base. A velocidade de inferência do modelo é de 45 FPS, atendendo aos requisitos em tempo real enquanto melhora a precisão da detecção.

关键词

nuvem de pontos LiDAR;detecção de objetos 3D;destilação de conhecimento;confiança na classificação

阅读全文

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

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