光伏电站航拍图像中组串级语义分割的人工智能算法

ARTIFICIAL INTELLIGENCE ALGORITHM FOR STRING-LEVEL SEMANTIC SEGMENTATION IN AERIAL IMAGES OF PHOTOVOLTAIC POWER STATION

  • 摘要: 光伏电站组串分割任务需要对物体边缘进行精确识别,从而得到精确的位置信息。针对此需求该文设计scSE-Unet8模型,将标点激励模块(scSE)引入U-Net模型并减少模型复杂度。在光伏电站航拍图像数据集上进行训练和验证。实验结果表明由于scSE模块对空间和通道特征进行重新修正,强调了重要的边缘特征信息,因此相比于其他模型scSE-Unet8对组串边缘像素检测效果更好。模型经过交叉验证后测试集上的mIoU(平均交并比)为98.62%。最后使用边界信息提取算法处理scSE-Unet8的输出结果,消除原分割结果中少量的误检和漏检,获得像素级别的组串边界。

     

    Abstract: The string segmentation task of photovoltaic power station requires to accurately identify the edge of PV string, so as to obtain accurate position information. For this task, this paper designs the scSE-Unet8 semantic segmentation model. The squeeze excitation module (SCSE) was introduced into the U-Net and the complexity of the model was reduced. The model was trained and verified on the aerial image data set of PV power station. The experimental results show that compared with the U-Net, scSE module revise the space and channel features, so as to emphasize the important edge feature information. Therefore, scSE-Unet8 has better effect on string edge pixel detection. After cross validation, the mIoU (Mean Intersection over Union) on the test set is 98.62%. The boundary information extraction algorithm was used to process the output result of scSE-Unet8, eliminate a small amount of false detection and missed detection in the original segmentation result, and the string boundary at the pixel level can be obtained.

     

/

返回文章
返回