一种三维目标检测模型的确定方法、装置及电子设备

    公开(公告)号:CN115359477A

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

    申请号:CN202211014882.8

    申请日:2022-08-23

    摘要: 本申请公开了一种三维目标检测模型的确定方法、装置及电子设备,该方法包括:获取训练样本;训练样本包括目标场景下的单目RGB样本图像及对应的标签信息;获取初始神经网络,初始神经网络包括特征提取模块、特征融合模块和检测头模块,特征提取模块包括多个基本特征提取单元,每个基本特征提取单元包括多个候选特征提取操作,多个候选特征提取操作的输出进行加权求和后,作为基本特征提取单元的输出;采用训练样本对初始神经网络进行训练,确定各个候选特征提取操作对应的权重,并得到初始检测模型;根据各个候选特征提取操作对应的权重及初始检测模型确定目标场景下的单目三维目标检测模型。本申请可自动搜索得到适合的单目三维目标检测模型。

    一种神经网络的量化方法、装置及电子设备

    公开(公告)号:CN115329957A

    公开(公告)日:2022-11-11

    申请号:CN202211029712.7

    申请日:2022-08-25

    IPC分类号: G06N3/08 G06N3/04

    摘要: 本申请公开了一种神经网络的量化方法、装置及电子设备,该方法包括:采用目标量化方式对目标神经网络的当前神经网络层的输入参数和网络参数进行量化;目标神经网络包括多个神经网络层;目标神经网络用于对图像中的目标进行识别;基于量化后的输入参数和当前神经网络层的量化后的网络参数,确定量化后的第一输出结果;基于输入参数及网络参数量化前的第一输出结果及量化后的第一输出结果对当前神经网络层的量化后的网络参数进行校正,得到量化后的当前神经网络层;确定量化后的当前神经网络层是否为目标神经网络的最后一个神经网络层,若是,则基于量化后的各个神经网络层,得到量化后的目标神经网络。如此可减少目标神经网络的参数量和计算量。