[发明专利]道路标牌信息识别方法、装置、计算机设备和存储介质

权利要求书

1.一种道路标牌信息识别方法，所述方法包括：

获取待识别道路标牌图像；

通过预设分类模型提取所述待识别道路标牌图像对应的分类结果以及所述分类结果对应的图像特征；

将所述图像特征输入到分类结果对应的预设信息识别模型进行信息识别，获取所述待识别道路标牌图像对应的道路标牌信息识别结果。

2.根据权利要求1所述的方法，其特征在于，所述通过预设分类模型提取所述待识别道路标牌图像对应的分类结果以及所述分类结果对应的图像特征之前，还包括：

获取模型训练数据，所述模型训练数据包括分类模型训练数据以及各分类类别的标牌信息识别模型训练数据；

通过所述分类模型训练数据对初始神经网络模型进行训练，获取预设分类模型，通过所述各分类类别的标牌信息识别模型训练数据对初始神经网络模型进行训练，获取各分类类别对应的预设信息识别模型。

3.根据权利要求2所述的方法，其特征在于，所述通过所述分类模型训练数据对初始神经网络模型进行训练，获取预设分类模型包括：

根据所述分类模型训练数据对初始神经网络模型进行训练，获取teacher模型；

通过知识蒸馏修改所述teacher模型中的softmax函数；

根据修改softmax函数后的teacher模型，获取student模型，将所述student模型作为预设分类模型。

4.根据权利要求1所述的方法，其特征在于，所述预设信息识别模型包括预设第一识别模型、预设第二识别模型以及预设第三识别模型；

当所述分类结果包括第一类型时，将所述第一类型对应的图像特征输入至预设第一识别模型，通过所述预设第一识别模型对图像特征中的信息进行识别，获取所述待识别道路标牌图像对应的第一类道路标牌信息识别结果，所述第一类型包括禁令、警告或指示；

当所述分类结果包括第二类型时，将所述第二类型对应的图像特征输入至预设第二识别模型，通过所述预设第二识别模型对图像特征中的信息进行识别，获取所述待识别道路标牌图像对应的第二类道路标牌信息识别结果，所述第二类型包括交通牌；

当所述分类结果包括第三类型时，将所述第三类型对应的图像特征输入至预设第三识别模型，通过所述预设第三识别模型对图像特征中的信息进行识别，获取所述待识别道路标牌图像对应的第三类道路标牌信息识别结果，所述第三类型包括信号灯；

归集已获取的道路标牌信息识别结果，得到所述待识别道路标牌图像对应的道路标牌信息识别结果。

5.根据权利要求4所述的方法，其特征在于，所述当所述分类结果包括第一类型时，将所述第一类型对应的图像特征输入至预设第一识别模型，通过所述预设第一识别模型对图像特征中的信息进行识别，获取所述待识别道路标牌图像对应的第一类道路标牌信息识别结果包括：

当所述分类结果包括第一类型时，通过所述预设第一识别模型对图像特征中的信息进行识别，获取细分类类别信息；

通过预设ocr文字识别模型识别所述图像特征中的数字信息；

根据所述细分类类别信息以及所述数字信息，获取所述待识别道路标牌图像对应的道路标牌信息识别结果。

6.根据权利要求4所述的方法，其特征在于，所述当所述分类结果包括第二类型时，将所述第二类型对应的图像特征输入至预设第二识别模型，通过所述预设第二识别模型对图像特征中的信息进行识别，获取所述待识别道路标牌图像对应的第二类道路标牌信息识别结果包括：

当所述分类结果包括第二类型时，通过所述预设第二识别模型对图像特征中的信息进行识别，获取交通牌类别信息；

通过预设ocr文字识别模型识别所述图像特征中的文字信息；

根据所述交通牌类别信息以及所述文字信息，获取所述待识别道路标牌图像对应的道路标牌信息识别结果。