技术特征:
1.一种基于人工智能的轮胎损伤检测方法,其特征在于,所述方法包括:采集带部件标签的车辆图像作为部件训练集,并基于所述部件训练集训练语义分割网络得到车辆部件分割网络,所述部件标签包括车辆图像中每一个像素点的车辆部件种类;采集目标车辆的实时车辆图像,将所述实时车辆图像输入所述车辆部件分割网络得到所述目标车辆的实时轮胎区域图像;搭建损伤检测第一网络;采集带损伤标签的轮胎区域图像作为损伤训练集,并基于所述损伤训练集和所述预设损失函数训练所述损伤检测第一网络得到损伤检测第二网络,所述损伤标签包括轮胎区域图像中所有损伤区域的区域信息和损伤种类;将所述实时轮胎区域图像输入所述损伤检测第二网络得到所述目标车辆的轮胎损伤检测结果。2.如权利要求1所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述采集目标车辆的实时车辆图像,将所述实时车辆图像输入所述车辆部件分割网络得到所述目标车辆的实时轮胎区域图像,包括:采集目标车辆的实时车辆图像,将所述实时车辆图像输入所述车辆部件分割网络得到实时车辆图像的部件分割图,所述实时车辆图像的部件分割图包括所述实时车辆图像中每一个像素点所属的车辆部件种类;将所述实时车辆图像的部件分割图中车辆部件种类为轮胎部件的像素点的像素值置为1,其他区域的像素值置为0,得到所述轮胎部件的掩码图,所述轮胎部件包括左前轮胎、左后轮胎、右前轮胎与右后轮胎;将所述轮胎部件的掩码图与所述实时车辆图像相乘,得到所述目标车辆的实时轮胎区域图像。3.如权利要求1所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述损伤检测第一网络为目标检测网络,所述搭建损伤检测第一网络包括:所述损伤检测第一网络的输入为轮胎区域图像,输出为所述轮胎区域图像的损伤检测结果,所述损伤检测结果包括所述轮胎区域图像中所有损伤区域的区域信息和损伤种类,所述区域信息为包含损伤区域的矩形框的宽高尺寸和中心点坐标,所述损伤种类包括缺失、穿孔缺失、开裂、严重损伤、鼓包;所述损伤检测第一网络包括一个编码器和多个参数共享的预测器;所述编码器对输入的轮胎区域图像不断进行特征提取得到下采样特征图,所述下采样特征图的数量为一个或多个;将所有下采样特征图输入多个参数共享的预测器中以获取每一张下采样特征图的子结果,其中所述预测器与所述下采样特征图一一对应,所述下采样特征图的子结果包括所述下采样特征图中所有损伤区域的区域信息和损伤种类;将所有下采样特征图的子结果作为所述轮胎区域图像的损伤检测结果。4.如权利要求1所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述基于所述损伤训练集和所述预设损失函数训练所述损伤检测第一网络得到损伤检测第二网络包括:基于所述损伤训练集和所述损伤检测第一网络构建第一预设损失函数;基于所述第一预设损失函数和目标检测损失函数构建预设损失函数,所述目标检测损
失函数与所述损伤检测第一网络的网络结构有关,所述预设损失函数满足关系式:loss=loss1+loss2其中,loss1为所述第一预设损失函数,loss2为所述目标检测损失函数,loss为预设损失函数;基于所述损伤训练集和所述预设损失函数训练所述损伤检测第一网络得到损伤检测第二网络。5.如权利要求4所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述基于所述损伤训练集和所述损伤检测第一网络构建第一预设损失函数包括:从所述损伤训练集中随机选取一张轮胎区域图像作为目标图像,将所述目标图像的损伤标签作为目标损伤标签,所述目标损伤标签包括所述目标图像中所有损伤区域的区域信息和损伤种类;将所述目标图像输入所述损伤检测第一网络得到所述目标图像的下采样特征图,所述下采样特征图为一个或多个;依据下采样特征图的尺寸将所述目标损伤标签映射到所有下采样特征图上,得到每一个下采样特征图的损伤标签,所述下采样特征图的损伤标签包括所述目标图像中所有损伤区域在该下采样特征图中的区域信息和损伤种类;基于所述下采样特征图的损伤标签构建每一个下采样特征图的特征差异矩阵,所有特征差异矩阵均为m行m列的方阵,其中m为所述目标图像中不同损伤种类的数量,所述下采样特征图的特征差异矩阵中第i行第j列的数值表示该下采样特征图中损伤种类i和损伤种类j的特征差异度,所述特征差异度满足关系式:其中,表示第u个下采样特征图中损伤种类i的区域信息中所有像素点的数量,表示第u个下采样特征图中损伤种类i的区域信息中第k
i
个像素点的像素值,表示第u个下采样特征图中损伤种类j的区域信息中所有像素点的数量,表示第u个下采样特征图中损伤种类j的区域信息中第k
j
个像素点的像素值,表示第u个下采样特征图中损伤种类i和损伤种类j的特征差异度;基于所有下采样特征图的特征差异矩阵和所述目标图像的标准差异矩阵构建第一预设损失函数,在所述标准差异矩阵中,对角线的数值为0,其他位置的数值为所述目标图像中不同损伤种类之间特征差异度的预设阈值。6.如权利要求5所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述依据下采样特征图的尺寸将所述目标损伤标签映射到所有下采样特征图上,得到每一个下采样特征图的损伤标签,所述下采样特征图的损伤标签包括所述目标图像中所有损伤区域在该下采样特征图中的区域信息和损伤种类,包括:从所述目标损伤标签中随机选取一个损伤区域的区域信息和损伤种类作为目标损伤
信息;依据映射模型和下采样特征图的尺寸将所述目标损伤信息映射至所有下采样特征图上以获取所述目标损伤信息的映射结果,所述映射模型满足关系式:其中,x
*
,y
*
,w
*
,h
*
表示所述目标损伤信息的区域信息,所述区域信息为包含损伤区域的矩形框,x
*
,y
*
为所述区域信息的中心点坐标,w
*
,h
*
为所述区域信息的宽高尺寸,z
*
表示所述目标损伤信息的损伤种类,w
*
,h
*
为所述目标图像的宽高尺寸,w
u
,h
u
为第u个下采样特征图的宽高尺寸,表示对参数x进行向上取整操作,(x
u
,y
u
,w
u
,h
u
,z
u
)为所述目标损伤信息映射至第u个下采样特征图上的映射结果;遍历所述目标损伤标签中所有损伤区域的损伤信息得到每一个损伤信息的映射结果,所述损伤信息包括损伤区域的区域信息和损伤种类;将同一下采样特征图上的所有映射结果作为所述下采样特征图的损伤标签,所述下采样特征图的损伤标签包括所述目标图像中所有损伤区域在该下采样特征图中的区域信息和损伤种类。7.如权利要求5所述的基于人工智能的轮胎损伤检测方法,其特征在于,所述第一预设损失函数满足关系式:其中,n为下采样特征图的数量,c
u
为第u个下采样特征图的特征差异矩阵;c
*
为所述目标图像的标准差异矩阵,‖c
u-c
*
‖0为l0范数,用于计算c
u-c
*
中所有非零数值的数量,loss1为第一预设损失函数。8.一种基于人工智能的轮胎损伤检测装置,其特征在于,所述装置包括:第一训练单元,用于采集带部件标签的车辆图像作为部件训练集,并基于所述部件训练集训练语义分割网络得到车辆部件分割网络,所述部件标签包括车辆图像中每一个像素点的车辆部件种类;分割单元,用于采集目标车辆的实时车辆图像,将所述实时车辆图像输入所述车辆部件分割网络得到所述目标车辆的实时轮胎区域图像;搭建单元,用于搭建损伤检测第一网络;第二训练单元,用于采集带损伤标签的轮胎区域图像作为损伤训练集,并基于所述损伤训练集和所述预设损失函数训练所述损伤检测第一网络得到损伤检测第二网络,所述损伤标签包括轮胎区域图像中所有损伤区域的区域信息和损伤种类;损伤检测单元,用于将所述实时轮胎区域图像输入所述损伤检测第二网络得到所述目标车辆的轮胎损伤检测结果。9.一种电子设备,其特征在于,所述电子设备包括:存储器,存储有计算机可读指令;及处理器,执行所述存储器中存储的计算机可读指令以实现如权利要求1至7中任意一项所述的基于人工智能的轮胎损伤检测方法。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机可读指令,所述计算机可读指令被处理器执行时实现如权利要求1至7中任一项所述的基于人工智能的轮胎损伤检测方法。
技术总结
本申请提出一种基于人工智能的轮胎损伤检测方法、装置、电子设备及存储介质,基于人工智能的轮胎损伤检测方法包括:采集带部件标签的车辆图像作为部件训练集,基于部件训练集训练语义分割网络得到车辆部件分割网络;采集目标车辆的实时车辆图像,将实时车辆图像输入车辆部件分割网络得到目标车辆的实时轮胎区域图像;搭建损伤检测第一网络;采集带损伤标签的轮胎区域图像作为损伤训练集,基于损伤训练集和预设损失函数训练损伤检测第一网络得到损伤检测第二网络;将实时轮胎区域图像输入损伤检测第二网络得到轮胎损伤检测结果,轮胎损伤检测结果包括实时轮胎区域图像中所有损伤区域的区域信息和损伤种类。本申请能提高轮胎损伤检测的准确度。损伤检测的准确度。损伤检测的准确度。
技术研发人员:方起明 刘莉红 刘玉宇 肖京
受保护的技术使用者:平安科技(深圳)有限公司
技术研发日:2022.07.28
技术公布日:2022/10/21