公开(公告)号：CN115359474A

公开(公告)日：2022-11-18

申请号：CN202210894321.5

申请日：2022-07-27

Applicant: 成都信息工程大学

Inventor： 郭本俊 ,  文立玉 ,  廖文龙 ,  赖育邦 ,  许源平 ,  俞文越 ,  田霞飞

IPC: G06V20/64 ,  G06V10/764 ,  G06V10/774 ,  G06V10/80 ,  G06V10/82

Abstract: 本发明公开一种适用于移动端的轻量级三维目标检测方法、装置及介质，所述方法包括：获取或制作用于3D目标检测的数据集；搭建模型，所述模型包括MobileViT block、三个分支和全连接层，所述MobileViT block与三个分支连接，各分支连接所述全连接层；基于所述数据集对搭建的模型进行训练；对训练后的模型进行处理得到onnx模型；将图像数据进行预处理并转换，输入至所述onnx模型中得到模型输出；基于模型输出，计算3D目标的全局方位角和3D检测框的多个顶点坐标，并绘制3D检测框和鸟瞰图。本发明同时拥有CNN和ViT相类似的属性和优点，以此可以用更少的参数学到更好的特征表示，从而实现移动端3D目标检测的任务。