应用光学, 2019, 40 (5): 786, 网络出版: 2019-11-05  

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

Aerial image de-roping based on U-net model
作者单位
1 武汉工程大学 湖北省视频图像与高清投影工程技术研究中心, 湖北 武汉 430073
2 深圳光启高等理工研究院, 广东 深圳 518000
摘要
光启云端号平台是用电缆绳索牵引气囊的, 由于空中航拍摄像系统悬挂在气囊下方, 摄取的图像不可避免地含有绳带信息, 这些绳带信息影响图像质量, 所以在场景分析和目标检测中需要剔除。提出一种基于U-net模型的绳带检测算法, 引入深度可分离卷积提高计算速度, 采用一种带权重的交叉熵作为损失函数, 解决类别不均衡带来的收敛不稳定问题, 最终的模型能够用较少的样本在较短的时间内, 快速准确地检测绳带, 利用快速行进修复算法(FMM)对绳带图像进行了修复。实验结果表明: 该算法的mIOU达到62.8%, 得到了较好的去绳结果。
Abstract
The Kuang-chi cloud platform is used to pull the airbag with a cable rope. Since the aerial photography system is suspended under the airbag, the captured image inevitably contains the rope information, which affects the image quality and needs to eliminated in the scene analysis and object detection. A rope detection algorithm based on U-net model was proposed, which introduced the separable convolution to improve the detection speed and adopted a weighted cross entropy as a loss function to solve the problem of convergence instability caused by category imbalance. The final model was able to quickly and accurately detect the rope with fewer data in less time , and then the rope image waseliminated using the fast marching method algorithm (FMM). The test results show that the mean intersection over union(mIOU) of the algorithm reaches 62.8% and has obtained good results.

洪汉玉, 孙建国, 栾琳, 王硕, 郑新波. 基于U-net模型的航拍图像去绳带方法[J]. 应用光学, 2019, 40(5): 786. HONG Hanyu, SUN Jianguo, LUAN Ling, WANG Shuo, ZHENG Xinbo. Aerial image de-roping based on U-net model[J]. Journal of Applied Optics, 2019, 40(5): 786.

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!