[发明专利]基于人工智能的车品牌识别方法、装置、设备及介质

权利要求书

1.一种基于人工智能的车品牌识别方法，其特征在于，所述基于人工智能的车品牌识别方法包括：

响应于接收到的待检测图像，对所述待检测图像进行resize处理，得到目标图像；

利用darknet53网络提取所述目标图像的车辆特征；

将所述车辆特征输入至预先训练的车品牌识别模型，输出第一特征图、第二特征图及第三特征图，包括：

对所述车辆特征执行第一卷积层运算，得到第一中间特征，并对所述第一中间特征执行第一联合卷积运算，得到所述第一特征图，其中，所述第一卷积层及所述第一联合卷积采用第一交并比训练而得到；

对所述第一中间特征执行上采样运算，得到第一特征；

对所述第一特征执行第二卷积层运算，得到第二中间特征，并对所述第二中间特征执行第二联合卷积运算，得到所述第二特征图，其中，所述第二卷积层及所述第二联合卷积采用第二交并比训练而得到；

对所述第二中间特征执行所述上采样运算，得到第二特征；

对所述第二特征执行第三卷积层运算及第三联合卷积运算，得到所述第三特征图，其中，所述第三卷积层及所述第三联合卷积采用第三交并比训练而得到；

其中，所述车品牌识别模型为采用多交并比及交并比损失训练YOLOv3网络而得到的级联网络结构；

获取所述车品牌识别模型的目标anchor box；

对于所述第一特征图、所述第二特征图及所述第三特征图中的每个特征图，利用所述目标anchor box在每个特征图上进行识别，输出与每个特征图对应的预测anchor box坐标、每个预测anchor box坐标的目标得分及车品牌的预测概率，从所述预测anchor box坐标中选择所述目标得分最高的预测anchor box坐标作为车辆的预测坐标；

将所述车辆的预测坐标映射到所述待检测图像上，得到映射图，输出所述映射图及所述车品牌的预测概率作为识别结果。

2.如权利要求1所述的基于人工智能的车品牌识别方法，其特征在于，所述基于人工智能的车品牌识别方法还包括：

当采用递进方式筛选样本时，所述第一交并比小于所述第二交并比，所述第二交并比小于所述第三交并比。

3.如权利要求1所述的基于人工智能的车品牌识别方法，其特征在于，所述基于人工智能的车品牌识别方法还包括：

获取训练样本；

将所述训练样本输入至YOLOv3网络，输出第一样本特征图、第二样本特征图及第三样本特征图；

对于所述第一样本特征图、所述第二样本特征图及所述第三样本特征图中的每个样本特征图，确定每个样本特征图对应的样本预测anchor box坐标及样本预测概率；

确定每个样本特征图的实际anchor box坐标、样本实际概率，及每个样本特征图对应的实际中心点坐标及实际宽高坐标；

根据所述样本预测anchor box坐标及所述实际anchor box坐标计算每个样本特征图的交并比损失；

根据每个样本特征图对应的样本预测anchor box坐标确定每个样本特征图对应的预测中心点坐标及预测宽高坐标；

根据每个样本特征图对应的预测中心点坐标及实际中心点坐标计算每个样本特征图的中心点坐标损失；

根据每个样本特征图对应的预测宽高坐标及实际宽高坐标计算每个样本特征图的宽高坐标损失；

根据每个样本特征图对应的样本预测概率及样本实际概率计算每个样本特征图的类别损失；

计算所述交并比损失、所述中心点坐标损失、所述宽高坐标损失及所述类别损失的和作为损失函数；

当所述损失函数的取值小于或者等于配置损失时，停止训练，得到所述车品牌识别模型。