[发明专利]一种太阳能光伏板识别方法、装置、电子设备及存储介质

权利要求书

1.一种太阳能光伏板识别方法，其特征在于，所述方法包括：

利用预处理后的验证样本对预设的语义分割模型进行验证，以优化模型参数；

利用预先获取的测试影像集和优化的所述语义分割模型进行模型预测，以获取所述测试影像集中每个像素点包含太阳能光伏板的概率；

根据所述概率和预设的概率阈值确定包含所述太阳能光伏板的像素点区域，以根据所述像素点区域获取所述太阳能光伏板的第一识别结果。

2.根据权利要求1所述的太阳能光伏板识别方法，其特征在于，在所述利用预处理后的验证数据对预设的语义分割模型进行验证，以优化模型参数的步骤之前，所述方法还包括：

构建所述语义分割模型；

获取训练样本并对所述语义分割模型进行训练。

3.根据权利要求2所述的太阳能光伏板识别方法，其特征在于，所述语义分割模型包括编码器和解码器，所述构建所述语义分割模型，包括：

所述编码器利用高分辨率网络提取太阳能光伏板特征；

利用所述解码器对所述太阳能光伏板特征所在的特征层进行重构，以获得所述太阳能光伏板的图像信息。

4.根据权利要求2所述的太阳能光伏板识别方法，其特征在于，所述获取训练样本并对所述语义分割模型进行训练，包括：

获取包含所述太阳能光伏板的第一遥感影像；

对所述第一遥感影像进行标注以获取对应的样本标签；

对带有所述样本标签的第一遥感影像进行剪裁以生成样本集，所述样本集包括按照预设比例进行随机分配的所述训练样本和所述验证样本；

设置训练超参数FocalLoss损失函数、CosineAnnealing学习率调整函数和Adam优化器并将所述训练样本输入所述语义分割模型进行训练。

5.根据权利要求1所述的太阳能光伏板识别方法，其特征在于，所述方法还包括：

获取包含所述太阳能光伏板的第二遥感影像；

对所述第二遥感影像进行随机抽取，以对抽取影像进行标注，生成标记影像，所述标记影像即为所述测试影像集；

利用所述标记影像对所述语义分割模型进行评估。

6.根据权利要求5所述的太阳能光伏板识别方法，其特征在于，所述利用所述标记影像对所述语义分割模型进行评估，包括：

利用所述标记影像和所述语义分割模型进行模型预测，以获取所述标记影像的第二识别结果；

将所述标记影像上的标记真值与所述第二识别结果基于mIOU评价指标进行比对，以获取评估结果；

所述mIOU评价指标表示为：

其中，P为每个标记影像的所述第二识别结果对应的预测值的像素的集合，G为所述标记真值对应的像素的集合，k为输入的测试影像即标记影像的个数，i取从1到k的整数。

7.一种太阳能光伏板识别装置，其特征在于，所述装置包括：

优化模块，用于利用预处理后的验证样本对预设的语义分割模型进行验证，以优化模型参数；

预测模块，用于利用预先获取的测试影像集和优化的所述语义分割模型进行模型预测，以获取所述测试影像集中每个像素点包含太阳能光伏板的概率；

识别模块，用于根据所述概率和预设的概率阈值确定包含所述太阳能光伏板的像素点区域，以根据所述像素点区域获取所述太阳能光伏板的第一识别结果。

8.根据权利要求7所述的太阳能光伏板识别装置，其特征在于，所述装置还包括：

构建模块，用于构建所述语义分割模型；

训练模块，用于获取训练样本并对所述语义分割模型进行训练。