基于U-net模型的航拍图像去绳带方法

作者:洪汉玉; 孙建国; 栾琳; 王硕; 郑新波
来源:应用光学, 2019, 40(05): 786-794.
DOI:10.5768/JAO201940.0502001

摘要

光启云端号平台是用电缆绳索牵引气囊的,由于空中航拍摄像系统悬挂在气囊下方,摄取的图像不可避免地含有绳带信息,这些绳带信息影响图像质量,所以在场景分析和目标检测中需要剔除。提出一种基于U-net模型的绳带检测算法,引入深度可分离卷积提高计算速度,采用一种带权重的交叉熵作为损失函数,解决类别不均衡带来的收敛不稳定问题,最终的模型能够用较少的样本在较短的时间内,快速准确地检测绳带,利用快速行进修复算法(FMM)对绳带图像进行了修复。实验结果表明:该算法的mIOU达到62.8%,得到了较好的去绳结果。

全文