摘要
本发明公开了一种基于知识蒸馏的车路协同3D目标检测方法及装置,属于自动驾驶感知领域。方法包括:获取预训练完成的教师模型;教师模型的输入为自车多模态数据和他车多模态数据;该多模态数据包括多视角二维图像数据和激光雷达点云数据;将教师模型中对激光雷达点云数据进行处理的相关模块去除,以利用剩余模块得到与教师模型架构相似的学生模型;学生模型的输入为多视角二维图像数据;利用知识蒸馏方法,以利用教师模型对学生模型进行监督训练,以利用训练完成的学生模型对自车进行3D目标检测。本发明能够在保持高检测精度的同时能够降低计算复杂度。
技术关键词
二维图像数据
激光雷达点云数据
教师
语义特征
学生
像素点
知识蒸馏方法
多视角
特征提取模块
多模态
特征提取器
点云特征
网络
计算机设备
关系