[发明专利]基于语义分割的图像分类方法、装置、设备及介质

说明书

本发明涉及人工智能技术领域，尤其涉及一种基于语义分割的图像分类方法、装置、设备及存储介质。该基于语义分割的图像分类方法包括获取待分类图像以及图像分类模型；将所述待分类图像输入至所述第一语义分割网络中进行语义分割处理，得到语义分割结果；其中，所述语义分割结果包括所述第一编解码层的解码输出以及分类特征图；所述分类特征图用于反映所述待分类图像中的图像属性信息；对所述分类特征图与所述解码输出进行拼接，得到拼接后的目标特征图；将所述目标特征图输入至所述第二语义分割网络中进行语义分割处理，得到所述待分类图像对应的分类结果。该方法可有效提高图像分类的准确度。

技术领域

本发明涉及人工智能技术领域，尤其涉及一种基于语义分割的图像分类方法、装置、设备及介质。

背景技术

随着自动、辅助驾驶的发展和普及，快速、准确的感知车辆周边的环境显得格外重要。车道线的分类识别是车辆感知系统中重要的组成部分，而车道线的类别信息对车辆行为的决策起着重要的作用。

目前车道线的分类识别大多采用语义分割的方法，即将待分类图像作为一个像素集，通过对集合中的每个像素分类，以得到车道线的类别信息。但该种车道线分类方法完全忽略了像素本身的底层特征，导致识别精度不高。

发明内容

本发明实施例提供一种基于语义分割的图像分类方法、装置、设备及介质，以解决当前基于语义分割的图像分类准确度不高的问题。

一种基于语义分割的图像分类方法，包括：

获取待分类图像以及图像分类模型；其中，所述图像分类模型包括级联的第一语义分割网络以及第二语义分割网络；所述第一语义分割网络包括依次连接的第一编解码层以及第一分类层；

将所述待分类图像输入至所述第一语义分割网络中进行语义分割处理，得到语义分割结果；其中，所述语义分割结果包括所述第一编解码层的解码输出以及所述第一分类层根据所述解码输出进行分类所得到的分类特征图；所述分类特征图用于反映所述待分类图像中的图像属性信息；

对所述分类特征图与所述解码输出进行拼接，得到拼接后的目标特征图；

将所述目标特征图输入至所述第二语义分割网络中进行语义分割处理，得到所述待分类图像对应的分类结果。

一种基于语义分割的图像分类装置，包括：

模型获取模块，用于获取待分类图像以及图像分类模型；其中，所述图像分类模型包括级联的第一语义分割网络以及第二语义分割网络；所述第一语义分割网络包括依次连接的第一编解码层以及第一分类层；

第一语义分割模块，用于将所述待分类图像输入至所述第一语义分割网络中进行语义分割处理，得到语义分割结果；其中，所述语义分割结果包括所述第一编解码层的解码输出以及所述第一分类层根据所述解码输出进行分类所得到的分类特征图；所述分类特征图用于反映所述待分类图像中的图像属性信息；

特征图拼接模块，用于对所述分类特征图与所述解码输出进行拼接，得到拼接后的目标特征图；

第二语义分割模块，用于将所述目标特征图输入至所述第二语义分割网络中进行语义分割处理，得到所述待分类图像对应的分类结果。

一种计算机设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现上述基于语义分割的图像分类方法的步骤。

一种计算机存储介质，所述计算机存储介质存储有计算机程序，所述计算机程序被处理器执行时实现上述基于语义分割的图像分类方法的步骤。