BDCN 和 U-net 边缘生成两阶段修复算法

李海燕¹,王伟华¹,郭磊¹,李海江²,李红松¹

1. 云南大学信息学院
2.云南交通投资建设集团有限公司

收稿日期:2020-12-18修回日期:2021-06-08出版日期:2021-10-28发布日期:2021-09-06
通讯作者:郭磊E-mail:guolei@ynu.edu.cn

基金资助:国家自然科学基金项目(61861045 ); 云南省****“ 教学名师冶 项目( 2019050057 ); 云南省基础研究计划重点项目(202101AS0070031)

Two-Stage Network Inpainting Algorithm Based on BDCN and U-net Edge Generation

Received:2020-12-18Revised:2021-06-08Online:2021-10-28Published:2021-09-06

摘要/Abstract

摘要： 为了对图像中大面积不规则缺失区域进行合理的结构修复和精细的纹理重构,提出了一种基于双向级联边缘检测网络(BDCN)和 U-net 残缺边缘生成的两阶段网络图像修复算法. 第一阶段首先基于 BDCN 网络提取边缘,然后基于 U-net 架构的边缘生成网络用下采样对缺失图像边缘提取特征,再结合上采样的输入信息和下采样各层信息还原图像边缘纹理细节;第二阶段使用空洞卷积进行下采样和上采样,经过残差网络重建细节丰富的缺失图像. 在公开数据集上将所提算法与现有经典算法进行对比,实验结果表明,所提算法能得到合理的结构和精细的纹理细节,优于对比算法.

中图分类号:
TP391

引用本文

李海燕 王伟华 郭磊 李海江 李红松. BDCN 和 U-net 边缘生成两阶段修复算法[J]. 北京邮电大学学报, 2021, 44(05): 121-126.





PDF全文下载地址:

https://journal.bupt.edu.cn/CN/article/downloadArticleFile.do?attachType=PDF&id=4825