Inproceedings,
Infrared image segmentation for photovoltaic panels based on res-unet
Chinese Conference on Pattern Recognition and Computer Vision (PRCV), 11857, page 611--622. Springer International Publishing, (2019)
DOI: 10.1007/978-3-030-31654-9_52
Abstract
Infrared image segmentation is the basis of error detection for photovoltaic panels. In this work, the infrared image data are collected by infrared thermal imager from the view of unmanned aerial vehicle (UAV). A semantic segmentation neural network named Deep Res-UNet, which combines the strengths of residual learning, transfer learning, and U-Net, is proposed for infrared image segmentation. Residual units are applied in both the encoding and decoding path, which makes the whole deep network ease to train. A modified ResNet-34 with pre-trained weights is utilized to get better feature representation. In the modified ResNet-34, maxpooling layer is removed for reducing the loss in resolution, an additional conv1 stage is added to copy features for the corresponding decoding path and the skip block with dilated residual block is added to generate features with larger resolution and larger receptive field. A new loss function combining Binary Cross-Entropy (BCE) and Dice is proposed to get better results. Additionally, Conditional Random Field (CRF) is integrated into the model as a post-process. The experimental results show that the prediction results of the proposed model are 97.11% and 94.47% respectively on the two evaluation indexes of F1 and Jaccard index, which is better than FCN-8s, SegNet, U-Net and two descendants of U-Net and ResNet: ResUnet and ResNet34-Unet.
