Infrared Image Segmentation for Photovoltaic Panels Based on Res-UNet
| Autor: | Shifen Zhou, Xianggong Hong, Qingcai Wang, Hao Zhang |
|---|---|
| Rok vydání: | 2019 |
| Předmět: |
Conditional random field
Artificial neural network Computer science business.industry 0211 other engineering and technologies Pattern recognition 02 engineering and technology Residual Feature (computer vision) Encoding (memory) 0202 electrical engineering electronic engineering information engineering 020201 artificial intelligence & image processing Segmentation Artificial intelligence business Decoding methods 021101 geological & geomatics engineering Block (data storage) |
| Zdroj: | Pattern Recognition and Computer Vision ISBN: 9783030316532 PRCV (1) |
| DOI: | 10.1007/978-3-030-31654-9_52 |
| Popis: | 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 \(F_1\) and Jaccard index, which is better than FCN-8s, SegNet, U-Net and two descendants of U-Net and ResNet: ResUnet and ResNet34-Unet. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|