[发明专利]目标识别方法及装置、电子设备及存储介质

说明书

本申请实施例提供一种目标识别方法及装置、电子设备及存储介质，其中，确定用于对对象的外观颜色进行分割的语义分割网络；其中，所述语义分割网络是采用样本图像集和外观标签库进行训练得到的；所述样本图像集中的样本对象已标注自身的外观颜色；所述外观标签库用于表征所述样本对象的类别与外观颜色之间的对应关系；获取包括待识别对象的待处理图像；采用所述语义分割网络，对所述待识别对象的外观颜色进行语义分割，确定所述待识别对象的目标类别。

技术领域

本申请实施例涉及图像处理技术领域，涉及但不限于一种目标识别方法及装置、电子设备及存储介质。

背景技术

近年来，共享单车成为新兴的城市交通产业，对于共享单车的管控已成为智慧城市的必需业务。在实际应用场景中，对共享单车进行识别时，由于受自然光的光照明暗、背景杂乱或视角场景变换等因素的影响，无法保证对共享单车的识别结果。

发明内容

本申请实施例提供一种目标识别技术方案。

本申请实施例的技术方案是这样实现的：

第一方面，本申请实施例提供一种目标识别方法，所述方法包括：

确定用于对对象的外观颜色进行分割的语义分割网络；其中，所述语义分割网络是采用样本图像集和外观标签库进行训练得到的；所述样本图像集中的样本对象已标注自身的外观颜色；所述外观标签库用于表征所述样本对象的类别与外观颜色之间的对应关系；

获取包括待识别对象的待处理图像；

采用所述语义分割网络，对所述待识别对象的外观颜色进行语义分割，确定所述待识别对象的目标类别。

第二方面，本申请实施例提供一种共享单车的颜色分割方法，所述方法包括：

确定用于对共享单车的外观颜色进行分割的语义分割网络；其中，所述语义分割网络是采用样本图像集和外观标签库进行训练得到的；所述样本图像集中的共享单车已标注自身的外观颜色；所述外观标签库用于表征所述共享单车的品牌与外观颜色之间的对应关系；

获取包括待识别共享单车的待处理图像；

采用所述语义分割网络，对所述待识别共享单车的外观颜色进行语义分割，确定所述待识别共享单车的目标品牌。

第三方面，本申请实施例提供一种目标识别装置，所述装置包括：

第一确定模块，用于确定用于对对象的外观颜色进行分割的语义分割网络；其中，所述语义分割网络是采用样本图像集和外观标签库进行训练得到的；所述样本图像集中的样本对象已标注自身的外观颜色；所述外观标签库用于表征所述样本对象的类别与外观颜色之间的对应关系；

第一获取模块，用于获取包括待识别对象的待处理图像；

第一分割模块，用于采用所述语义分割网络，对所述待识别对象的外观颜色进行语义分割，确定所述待识别对象的目标类别。

在上述装置中，所述语义分割网络包括用于对输入的图像进行特征提取的第一网络，和用于对所述第一网络提取的图像特征进行语义分割的第二网络，所述第一分割模块，包括：

第一提取子模块，用于采用所述第一网络对所述待处理图像进行特征提取，得到图像特征，并将所述图像特征输入所述第二网络；

第一分割子模块，用于采用所述第二网络，按照所述外观标签库，对所述图像特征进行语义分割，确定所述待识别对象的目标类别。

在上述装置中，所述装置还包括：

第二获取模块，用于获取包括样本对象的样本图像集合；

第二确定模块，用于确定所述样本对象所属的类别和所述样本对象的外观颜色；其中，所述外观颜色用于表征所述样本对象的主体颜色；