[发明专利]图像类别的识别方法、装置和电子设备

权利要求书

1.一种图像类别的识别方法，包括：

获取光谱图像，其中，所述光谱图像包括待识别的第一像素点和每个类别对应的标记为样本的第二像素点；

基于所述光谱图像对图像识别模型进行训练，由所述图像识别模型获取每个像素点的光谱语义特征、每个像素点与每个类别的最小距离、每个像素点的第一光谱与每个类别的第二光谱之间的光谱距离，对所述光谱语义特征、所述最小距离和所述光谱距离进行拼接，得到拼接特征，并基于所述拼接特征进行分类识别，输出每个像素点在每个类别下的识别概率；

基于所述第二像素点的识别概率，确定所述图像识别模型的损失函数，并基于所述损失函数调整所述图像识别模型，并返回基于所述光谱图像继续对调整后的图像识别模型进行训练，直至训练结束生成目标图像识别模型；

从所述目标图像识别模型输出的所述第一像素点在每个类别下的识别概率中识别最大识别概率，并将所述最大识别概率对应的类别确定为所述第一像素点对应的目标类别；

其中，获取所述每个像素点与每个类别的最小距离，包括：

获取任一像素点，获取所述任一像素点与每个类别所包含的每个第二像素点之间的第一距离；

针对任一类别，获取所述任一类别的所述第一距离中的最小值作为所述任一像素点与该类别的所述最小距离；

获取所述每个像素点的第一光谱与每个类别的第二光谱之间的光谱距离，包括：

将每个类别所包含的每个第二像素点的第一光谱作为该类别的第二光谱；

获取每个像素点的第一光谱与每个类别的第二光谱的平均值之间的向量距离，将所述向量距离作为所述光谱距离；

所述对所述光谱语义特征、所述最小距离和所述光谱距离进行拼接，得到拼接特征，包括：

对所述光谱语义特征、所述最小距离和所述光谱距离进行横向拼接，得到拼接特征。

2.根据权利要求1所述的方法，其中，所述获取每个像素点的光谱语义特征，包括：

将所述光谱图像输入所述图像识别模型的语义提取层，基于所述语义提取层对每个像素点的光谱进行语义特征提取，得到所述光谱语义特征。

3.根据权利要求1所述的方法，其中，所述获取每个像素点的第一光谱与每个类别的第二光谱的平均值之间的向量距离，包括：

对所述每个像素点的第一光谱进行降维处理，得到第一降维光谱；

对所述每个类别的第二光谱的平均值进行降维处理，得到第二降维光谱；

获取所述第一降维光谱和所述第二降维光谱之间的向量距离。

4.根据权利要求3所述的方法，其中，所述方法还包括：

对光谱进行主成分分析PCA处理，从所述光谱中提取主要特征分量，并基于所述主要特征分量，生成降维光谱；其中，所述光谱包括第一光谱和所述第二光谱，所述降维光谱包括所述第一降维光谱和所述第二降维光谱；或者，

获取光谱所对应的波段，对所述波段进行筛选，保留目标波段，并基于保留的目标波段上的光谱，生成降维光谱。