买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：深圳市锐明像素科技有限公司

摘要：本申请实施例公开了一种图像中道路积水的识别方法、装置、终端和存储介质；本申请实施例包括：获取输入图像；通过预设道路积水特征提取模型，从所述输入图像中提取得到积水特征图；通过分割头网络对所述积水特征图进行处理，得到输入图像对应的分割结果；通过分类头网络对所述积水特征图进行处理，得到输入图像对应的分类结果；基于所述分割结果以及所述分类结果，生成目标识别图。本申请实施例可以通过神经网络模型对道路图像的信息进行充分提取，更准确地将道路图像中可能存在的道路积水部分识别出来。

主权项：1.一种图像中道路积水的识别方法，其特征在于，包括：获取输入图像；通过预设道路积水特征提取模型，从所述输入图像中提取得到积水特征图；通过分割头网络对所述积水特征图进行处理，得到输入图像对应的分割结果；通过分类头网络对所述积水特征图进行处理，得到输入图像对应的分类结果；基于所述分割结果以及所述分类结果，生成目标识别图，其中，所述目标识别图用于表征所述输入图像是否包括道路积水；所述预设道路积水特征提取模型包括茎部网络、第一特征提取网络、第二特征提取网络以及第三特征提取网络，所述茎部网络、所述第一特征提取网络、所述第二特征提取网络以及所述第三特征提取网络依次连接，所述通过预设道路积水特征提取模型，从所述输入图像中提取得到积水特征图，包括：将所述输入图像输入至所述茎部网络，得到第一子特征图；通过所述第一特征提取网络，根据所述第一子特征图生成第二子特征图；对所述第一子特征图进行上采样处理，得到第三子特征图，其中，所述第三子特征图与所述积水特征图尺寸大小相同；通过所述第二特征提取网络，根据所述第二子特征图生成第四子特征图，其中，所述第二特征提取网络的网络结构与所述第一特征提取网络的网络结构相同；对所述第二子特征图进行上采样处理，得到第五子特征图，其中，所述第五子特征图与所述积水特征图尺寸大小相同；通过所述第三特征提取网络，根据所述第四子特征图生成第六子特征图；对所述第四子特征图进行上采样处理，得到第七子特征图，其中，所述第七子特征图与所述积水特征图尺寸大小相同；对所述第六子特征图进行上采样处理，得到第八子特征图，其中，所述第八子特征图与所述积水特征图尺寸大小相同；对所述第三子特征图、所述第五子特征图、所述第七子特征图以及所述第八子特征图进行特征融合，得到所述积水特征图；所述基于所述分割结果以及所述分类结果，生成目标识别图，包括：基于预设激活函数，对所述分割结果进行处理，得到单通道特征图；计算所述分类结果以及所述单通道特征图的乘积，得到所述目标识别图。

全文数据：

权利要求：

百度查询： 深圳市锐明像素科技有限公司 图像中道路积水的识别方法、装置、终端和存储介质