基于改进PReNet的轻量级单幅图像去雨算法研究

来源 :重庆大学 | 被引量 : 0次 | 上传用户:cr_idealism
下载到本地 , 更方便阅读
声明 : 本文档内容版权归属内容提供方 , 如果您对本文有版权争议 , 可与客服联系进行内容授权或下架
论文部分内容阅读
单幅图像去雨是计算机视觉领域的难题之一。在自然环境中,雨纹呈现不同尺度和不同方向的特点,增加了图像去雨难度。PReNet(Progressive Image Deraining Networks)算法作为目前去雨性能较好的轻量级单幅图像去雨算法,适用于实际存储资源受限的场合,但其尚存在雨纹去除不彻底和无法精细恢复图像背景细节等问题。因此,对PReNet算法进行改进,以保持其模型轻量的同时提升其去雨性能来满足实际应用需求,具有重要的理论和实际意义。论文对PReNet算法的原理进行了深入分析,围绕网络剪枝、多尺度特征提取、通道特征关联性和网络训练方式四个方面进行改进研究,构建了轻量级单幅图像去雨算法PRe Dense Net。论文的主要工作和贡献如下:1针对PReNet算法去除雨纹不彻底和无法精细恢复背景细节问题,改进并构建了轻量级去雨算法PReNet_D。首先引入空洞卷积增大PReNet算法感受野,在不增加模型复杂度情况下获取更多上下文信息,可提取多尺度雨纹特征;进而针对空洞卷积引起的无法有效获取多尺度信息和损失信息连续性问题,利用混合空洞卷积可有效提取多尺度物体信息性能,帮助去除雨纹;最后,在训练时引入感知损失,让去雨图像与参考无雨图的高层信息更加接近,以提高生成图像质量。2受到PReNet算法多阶段逐步去雨和模块间参数共享思想的启发,针对PReNet_D算法特征提取模块会影响信息在网络中传播的问题,改进并构建了轻量级单幅图像去雨算法PRe Dense Net。首先使用轻量级残差密集连接块作为特征提取模块,去除其批归一化和平均池化操作来减少模型参数,并引入空洞卷积和混合空洞卷积来有效提取多尺度雨纹信息,提升雨纹去除效果;其次,针对通道级特征缺乏关联性问题,引入压缩-激励模块来自适应学习各通道的权重值,从而提升去雨性能;此外,针对各阶段之间相关性问题,引入更轻量的GRU(Gate Recurrent Unit)来增强网络的长期记忆能力,前面阶段提取的重要特征可帮助后续阶段有效去雨;最后,在网络训练方面,使用像素级损失函数MSE(Mean Squared Error)和感知损失函数作为整体的损失函数来训练网络,帮助恢复图像背景细节。综合所有改进,形成一种轻量级单幅图像去雨算法PRe Dense Net。在Rain100H、Rain100L和Rain12三个合成数据集以及真实有雨图片上的实验结果表明,所提算法在保证模型轻量的同时,可以取得更好的去雨性能以及视觉上更清晰的去雨效果。
其他文献
滚动轴承作为一种广泛应用各种旋转机械中的通用零部件,是机械设备发生故障最高的原因之一。旋转机械的平稳运行完全取决于滚动轴承的健康状态,滚动轴承失效占这些机械设备故障的45-55%,其中轴承的磨损是最为常见和不易识别的故障类型,故有必要对轴承磨损故障实施状态监测。但是滚动轴承一般应用于复杂和恶劣的环境中,磨损信号在强噪声环境下相对比较微弱,不易于采集;且滚动轴承振动信号比较复杂,包含较大程度的机械本
学位
国家“十四五”规划将自动驾驶汽车作为了汽车行业重点方向之一。而复杂交通场景下的驾驶安全问题是当前制约自动驾驶技术在汽车行业大规模应用的关键问题。场地测试装备是检验和提升自动驾驶汽车安全性能、提升自动驾驶汽车研发过程效率的重要手段。汽车的自动驾驶级别越高,测试需要覆盖的场景越丰富。目前高端的自动驾驶汽车封闭场地测试系统技术皆被国外公司垄断,相较于国外厂家,我国尚未自主开发出一套功能齐全、精度高的自动
学位
城市快速路车道变窄区是一种常见的交通固定瓶颈,一般由道路设计、施工以及交通事故等原因造成。当瓶颈上游交通需求超过瓶颈区的最大通行能力时,常常导致瓶颈位置发生交通拥堵。而交通拥堵会导致瓶颈区通行能力下降,进一步加剧交通拥堵,严重影响交通运行效率。可变限速策略作为缓解瓶颈区交通拥堵的有效手段,通过调整瓶颈上游道路限速来限制进入瓶颈位置的车流量,可以抑制瓶颈区通行能力下降程度,从而缓解瓶颈区的交通拥堵。
学位
多移动机器人在国防、航空航天、运输等领域应用广泛。其中,多移动机器人一致性控制是一个核心问题,即要求设计一致性控制算法使系统的状态或输出达成一致。多机器人系统具有不确定且受到外部不确定干扰,因此研究不确定非线性多机器人系统的分布式一致性跟踪控制具有实际意义。为增强多机器人系统适应环境的能力,本文采用一种积分强化学习的方法,研究跟随者存在通信时滞、外部扰动以及输入约束下的最优输出协调跟踪控制问题,设
学位
航空航天领域的发展日新月异,产业的广度和深度不断扩展,对于航天火工品的需求也相应增加。装压药作为航天火工品制造过程中的关键环节,对成品整体的质量控制举足轻重,因此实现对装压药制造过程进行分析和控制有着重要的现实意义。而装压药的制造具有多品种、离散化、小批量、多批次的特点,制造过程中的偏差呈现出波动性、传递性和耦合性,进而给偏差源诊断带来不确定性。基于此背景,本课题开展航天火工品装压药制造质量控制与
学位
随着社会的高速发展,企业对外交易产生了大量银行回单凭证,应用于企业内部做账及税务核算等场景。银行回执单需要被快速识别和录入,而在识别过程中存在几个问题。其一,当银行回执单图像存在污渍、噪音干扰、文本行倾斜的极端情况下时,其识别的准确度较差;其二,银行回执单图像整体的识别效果存在一定的改进空间。论文基于上述问题,在文本识别阶段采用并改进了CRNN模型,在CRNN模型的卷积层和循环层分别融合不同的残差
学位
航天安全同国家利益息息相关,太空资源具有巨大的民用价值和商业价值,整流罩内环境温湿度保障作为火箭内部载荷安全保障的重要组成部分,对卫星设备安全具有重大影响。本文以某运载火箭整流罩内环境地面保障过程为研究背景,主要工作如下:针对特殊工况下罩内温湿度反馈丢失问题,建立了基于滑窗数据重组结合贝叶斯优化的Light GBM回归模型。通过对历史罩内温湿度数据和送风温湿度数据进行了预处理,再利用滑动窗口实现了
学位
齿轮箱是各种旋转机械设备中最为重要的部件之一,当其出现故障时设备将无法正常工作,有着很大的安全风险,因此对齿轮箱进行准确地故障诊断是十分有现实意义的。本文以直升机为背景,选取齿轮箱中齿轮作为研究对象,基于振动信号,从信号处理、特征提取以及故障分类等方面进行了探究,具体的工作内容如下:首先对齿轮箱中最常见的故障类型进行了介绍,探究了几种故障产生的原因以及齿轮箱的振动机理,从动力学的角度说明了齿轮振动
学位
随着中医药行业的规范化发展,散袋小包装中药以其包装上的优势逐渐颠覆了传统散装中药的主导地位。然而,不同于西药的硬质外包装,散袋小包装中药大多是柔性袋装药包,不易于抓取和识别,导致大部分中药房依然采用人工的方式来抓取并计量药包。实现散袋小包装中药自动出药,能够提高中药房的出药效率,降低出药错误率,既减轻了药剂师的负担,也减少了患者等待取药的时间。自动化出药的过程需要称量系统与真空吸盘阵列抓药系统的配
学位
随着智能汽车的发展,在实现全自动驾驶之前,智能汽车将长期处于驾驶员和驾驶自动化系统共同参与车辆控制的人机共驾阶段,使得如何提升人机协同效果受到了广泛的关注。此外,在交通密度日益增大的背景下,车辆之间的相互作用关系愈发复杂,车车协同对于改善人机协同驾驶车辆的控制性能有着不可忽视的作用。因此,如何一体化考虑智能汽车内部人机协同与外部车车协同问题,对于提高自动驾驶车辆通行效率,保证道路交通安全性有着重要
学位