CN109993804A - 一种基于条件生成对抗网络的道路场景去雾方法 - Google Patents
一种基于条件生成对抗网络的道路场景去雾方法 Download PDFInfo
- Publication number
- CN109993804A CN109993804A CN201910224071.2A CN201910224071A CN109993804A CN 109993804 A CN109993804 A CN 109993804A CN 201910224071 A CN201910224071 A CN 201910224071A CN 109993804 A CN109993804 A CN 109993804A
- Authority
- CN
- China
- Prior art keywords
- picture
- road scene
- true
- network based
- confrontation network
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000000034 method Methods 0.000 title claims abstract description 18
- 238000012549 training Methods 0.000 claims abstract description 10
- 239000003595 mist Substances 0.000 claims abstract description 9
- 238000012545 processing Methods 0.000 claims abstract description 8
- 238000012360 testing method Methods 0.000 claims abstract description 4
- 238000003475 lamination Methods 0.000 claims description 6
- 239000000203 mixture Substances 0.000 claims description 4
- 238000005070 sampling Methods 0.000 claims description 3
- 230000006870 function Effects 0.000 description 6
- 230000000694 effects Effects 0.000 description 5
- 238000011160 research Methods 0.000 description 4
- 238000013135 deep learning Methods 0.000 description 3
- 238000011161 development Methods 0.000 description 2
- 238000012544 monitoring process Methods 0.000 description 2
- 239000000126 substance Substances 0.000 description 2
- 238000004364 calculation method Methods 0.000 description 1
- 238000006243 chemical reaction Methods 0.000 description 1
- 238000004891 communication Methods 0.000 description 1
- 230000006866 deterioration Effects 0.000 description 1
- 235000013399 edible fruits Nutrition 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 238000011156 evaluation Methods 0.000 description 1
- 238000000605 extraction Methods 0.000 description 1
- 238000001914 filtration Methods 0.000 description 1
- 238000000926 separation method Methods 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/90—Determination of colour characteristics
Abstract
本发明涉及一种基于条件生成对抗网络的道路场景去雾方法,包括以下步骤:1)采用RESIDE数据集构建训练集和测试集;2)将训练集中的有雾图片x作为生成器G的输入并生成G(x),即无雾图片3)将生成的无雾图片发送给判别器D,判断图片真假;4)迭代设定轮次后,得到最优生成模型;5)将得到的最优生成模型应用于真实有雾图像,进行去雾处理。与现有技术相比,本发明具有鲁棒性好、处理时间短等优点。
Description
技术领域
本发明涉及图像处理技术领域,尤其是涉及一种基于条件生成对抗网络的道路场景去雾方法。
背景技术
随着计算机视觉的发展及其在交通、安全监控领域的应用,图像去雾已成为计算机视觉的重要研究领域。在雾霾天气导致能见度较低的恶劣气象条件下,由摄像机采集到的图片受大气中悬浮颗粒物(如雾、霾等)的影响,导致图片质量不佳,难以辨别图片中的物体特征,甚至影响例如室外监控、目标识别和交通导航中图片的质量。因此,有雾图像特征清晰化有着重要的研究意义。
目前,图像去雾的方法主要有两类:图像增强和模型去雾。图像增强类算法直接从图像处理角度出发,通过增强含雾图像的对比度,突出图片的特征或有效信息,一定程度上改善图片的视觉效果。但是该类方法忽略了图像降质的真正原因,所以对于场景复杂的图片无法提高图片的质量,甚至可能丢失图像的某些信息。常用的图像增强类算法包括直方图均衡、多尺度Retinex、同态滤波、Tan等。Kim等和Stark提出局部直方图均衡法,其主要是定义一个图像的子块,从而确定其直方图,再对该子块进行直方图均衡,子块的中心灰度被替换为直方图均衡后的灰度,如此根据每个像素的邻域对像素进行处理,有利于突出图像特征。多尺度Retinex算法将有雾图像中的照射分量和反射分量分离,消除以雾为主的照射分量对图像的影响,达到去雾的效果。但是在用该算法进行图像增强时,要计算照度分量,这在数学上是个欠定问题,只能通过近似估计。
模型去雾类算法通过建立大气散射模型,研究图像退化的物理原理,得出大气中悬浮颗粒物对光的散射作用以及对图片的影响,复原出更真实的图片,且在复杂场景中去雾效果较好,图像信息较完整。常用的模型去雾类算法有Tarel、Fattal、He等。Nayer等把大气对景物反射的光线的影响分为大气对景物光线的衰减和环境光的叠加,并分别进行推导,从而在根本上去雾,得到原始的无雾图片,且丢失的信息较少。He通过对大量无雾图像统计特征的观察,提出了暗道先验(Dark Channel Prior,DCP)的算法,其认为雾的浓度近似于最暗通道的数值,即在非天空的清晰区域,在RGB三个通道中有一个颜色通道的亮度很低甚至接近于0。该算法对非天空区域的图片有着良好的去雾效果,而对于有天空的这类的亮区域结果并不理想,且该算法计算量太大,效率较低。
近年来,随着深度学习(Deep Learning)的不断发展,其在各领域取得了较多的成果,其中也包含了对去雾的研究。但是由于数据的的缺失,这样的基于深度学习的方法并不多见。
发明内容
本发明的目的就是为了克服上述现有技术存在的缺陷而提供一种基于条件生成对抗网络的道路场景去雾方法。
本发明的目的可以通过以下技术方案来实现:
一种基于条件生成对抗网络的道路场景去雾方法,用以降低雾霾天气对道路场景获得图像的干扰,包括以下步骤:
1)采用RESIDE数据集构建训练集和测试集;
2)将训练集中的有雾图片x作为生成器G的输入并生成G(x),即无雾图片
3)将生成的无雾图片发送给判别器D,判断图片真假;
4)迭代设定轮次后,得到最优生成网络;
5)将得到的最优生成网络应用于真实有雾图像,进行去雾处理。
所述的步骤2)中,生成器G包括第一编码器、转换器和解码器。
所述的第一编码器由3层卷积层构成,用以提取输入图像的特征向量,所述的转换器由6层的ResNet模块构成,用以将源域中图片的特征向量转换到目标域中的特征向量,所述的解码器由3层反卷积层构成,用以从特征向量中还原得到低级特征。
所述的步骤3)中,判别器D由第二编码器组成,该第二编码器由4层反卷积层构成,实现对图片的下采样,得到示意图片真假的概率。
所述的步骤3)中,判别器D采用PatchGAN结构,该PatchGAN结构用以对图片70×70的patch进行判别。
所述的步骤4)中,网络训练的目标函数为:
其中,L(G,D)为损失函数,D(x,y)为真实图像的得分,D(x,G(x))为生成图像的得分,为所有真实图像的期望值,为所有生成图像的期望值。
与现有技术相比,本发明具有以下优点:
一、由于本发明采用生成对抗网络技术,能够更好地获取到雾天场景的特征,因此对多种雾天场景,如雾天街道场景,雾天的公园等,具有更好的鲁棒性。
二、本发明由于引入了ResNet模块,取得了更好的去雾效果,并且采用了PatchGAN结构的判别器,仅对图片70×70的patch进行判别,而不是整张图片,这能够减少图像处理时间。
附图说明
图1为条件生成对抗网络的生成器结构。
图2为条件生成对抗网络的判别器结构。
具体实施方式
下面结合附图和具体实施例对本发明进行详细说明。
实施例
下面结合附图,通过实例进一步描述本发明,但不以任何方式限制本发明范围。
本发明以RESIDE数据集为例,具体操作如下:
步骤一:采用RESIDE数据集,其中包含训练集和测试集;
步骤二:将训练集中的有雾图片x作为输入传送给生成器G生成G(x),即无雾图片
步骤三:将生成的无雾图片传送给判别器D,判断图片真假;
步骤四:迭代200轮后,得到最优生成模型;
步骤五:将得到的最优生成模型用于真实有雾图像,进行去雾处理。
在本实施中,生成器由三个部分组成:编码器、转换器和解码器;编码器由3层卷积层组成,实现输入图像特征的提取;转换器由6层的ResNet模块构成,实现将源域中图片的特征向量转换到目标域中的特征向量;解码器由3层反卷积层构成,实现从特征向量中还原出低级特征。
在本实施中,判别器仅由编码器组成;编码器为4层反卷积层,实现对图片不断的下采样,最终得到一个概率示意图片真假;判别器采用PatchGAN结构;PatchGAN结构仅对图片70×70的patch进行判别,而不是整张图片,这样本发明在得到相同去雾效果的条件下,使得图像处理时间更短。
在本实施中,网络结构的损失函数为:
目标函数中,生成器G要最小化该目标函数,判别器D要最大化该目标函数:
经过上述步骤,得到一个最优的生成模型,可以用来对其他场景的有雾图片进行去雾。
Claims (6)
1.一种基于条件生成对抗网络的道路场景去雾方法,用以降低雾霾天气对道路场景获得图像的干扰,其特征在于,包括以下步骤:
1)采用RESIDE数据集构建训练集和测试集;
2)将训练集中的有雾图片x作为生成器G的输入并生成G(x),即无雾图片
3)将生成的无雾图片发送给判别器D,判断图片真假;
4)迭代设定轮次后,得到最优生成网络;
5)将得到的最优生成网络应用于真实有雾图像,进行去雾处理。
2.根据权利要求1所述的一种基于条件生成对抗网络的道路场景去雾方法,其特征在于,所述的步骤2)中,生成器G包括第一编码器、转换器和解码器。
3.根据权利要求2所述的一种基于条件生成对抗网络的道路场景去雾方法,其特征在于,所述的第一编码器由3层卷积层构成,用以提取输入图像的特征向量,所述的转换器由6层的ResNet模块构成,用以将源域中图片的特征向量转换到目标域中的特征向量,所述的解码器由3层反卷积层构成,用以从特征向量中还原得到低级特征。
4.根据权利要求1所述的一种基于条件生成对抗网络的道路场景去雾方法,其特征在于,所述的步骤3)中,判别器D由第二编码器组成,该第二编码器由4层反卷积层构成,实现对图片的下采样,得到示意图片真假的概率。
5.根据权利要求1所述的一种基于条件生成对抗网络的道路场景去雾方法,其特征在于,所述的步骤3)中,判别器D采用PatchGAN结构,该PatchGAN结构用以对图片70×70的patch进行判别。
6.根据权利要求1所述的一种基于条件生成对抗网络的道路场景去雾方法,其特征在于,所述的步骤4)中,网络训练的目标函数为:
G*,
其中,L(G,D)为损失函数,D(x,y)为真实图像的得分,D(x,G(x))为生成图像的得分,为所有真实图像的期望值,为所有生成图像的期望值。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910224071.2A CN109993804A (zh) | 2019-03-22 | 2019-03-22 | 一种基于条件生成对抗网络的道路场景去雾方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910224071.2A CN109993804A (zh) | 2019-03-22 | 2019-03-22 | 一种基于条件生成对抗网络的道路场景去雾方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN109993804A true CN109993804A (zh) | 2019-07-09 |
Family
ID=67130920
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201910224071.2A Pending CN109993804A (zh) | 2019-03-22 | 2019-03-22 | 一种基于条件生成对抗网络的道路场景去雾方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN109993804A (zh) |
Cited By (11)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110675328A (zh) * | 2019-08-02 | 2020-01-10 | 北京巨数数字技术开发有限公司 | 基于条件生成对抗网络的低照度图像增强方法及装置 |
CN111179202A (zh) * | 2019-12-31 | 2020-05-19 | 内蒙古工业大学 | 一种基于生成对抗网络的单幅图像去雾增强方法和系统 |
CN111325337A (zh) * | 2019-08-19 | 2020-06-23 | 大连理工大学 | 一种基于自编码神经网络与最优传输的生成模型 |
CN111539885A (zh) * | 2020-04-21 | 2020-08-14 | 西安交通大学 | 一种基于多尺度网络的图像增强去雾方法 |
CN111709888A (zh) * | 2020-06-03 | 2020-09-25 | 中科九度(北京)空间信息技术有限责任公司 | 一种基于改进的生成对抗网络的航拍图像去雾方法 |
CN111833306A (zh) * | 2020-06-12 | 2020-10-27 | 北京百度网讯科技有限公司 | 缺陷检测方法和用于缺陷检测的模型训练方法 |
CN111882495A (zh) * | 2020-07-05 | 2020-11-03 | 东北林业大学 | 一种基于自定义模糊逻辑与gan的图像高光处理方法 |
CN112308156A (zh) * | 2020-11-05 | 2021-02-02 | 电子科技大学 | 一种基于对抗学习的两阶段图像变化检测方法 |
CN113450261A (zh) * | 2020-03-25 | 2021-09-28 | 江苏翼视智能科技有限公司 | 一种基于条件生成对抗网络的单幅图像去雾方法 |
CN113554872A (zh) * | 2021-07-19 | 2021-10-26 | 昭通亮风台信息科技有限公司 | 一种位于交通路口及弯道的检测预警方法及系统 |
CN113724156A (zh) * | 2021-08-09 | 2021-11-30 | 浙江工业大学 | 一种结合大气散射模型的生成对抗网络去雾方法和系统 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106127702A (zh) * | 2016-06-17 | 2016-11-16 | 兰州理工大学 | 一种基于深度学习的图像去雾算法 |
CN108615226A (zh) * | 2018-04-18 | 2018-10-02 | 南京信息工程大学 | 一种基于生成式对抗网络的图像去雾方法 |
CN109493303A (zh) * | 2018-05-30 | 2019-03-19 | 湘潭大学 | 一种基于生成对抗网络的图像去雾方法 |
-
2019
- 2019-03-22 CN CN201910224071.2A patent/CN109993804A/zh active Pending
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106127702A (zh) * | 2016-06-17 | 2016-11-16 | 兰州理工大学 | 一种基于深度学习的图像去雾算法 |
CN108615226A (zh) * | 2018-04-18 | 2018-10-02 | 南京信息工程大学 | 一种基于生成式对抗网络的图像去雾方法 |
CN109493303A (zh) * | 2018-05-30 | 2019-03-19 | 湘潭大学 | 一种基于生成对抗网络的图像去雾方法 |
Non-Patent Citations (1)
Title |
---|
P. SUAREZ AT EL: "Deep Learning Based Single Image Dehazing", 《2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW)》 * |
Cited By (19)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110675328A (zh) * | 2019-08-02 | 2020-01-10 | 北京巨数数字技术开发有限公司 | 基于条件生成对抗网络的低照度图像增强方法及装置 |
CN110675328B (zh) * | 2019-08-02 | 2023-04-28 | 北京巨数数字技术开发有限公司 | 基于条件生成对抗网络的低照度图像增强方法及装置 |
CN111325337A (zh) * | 2019-08-19 | 2020-06-23 | 大连理工大学 | 一种基于自编码神经网络与最优传输的生成模型 |
CN111179202A (zh) * | 2019-12-31 | 2020-05-19 | 内蒙古工业大学 | 一种基于生成对抗网络的单幅图像去雾增强方法和系统 |
CN111179202B (zh) * | 2019-12-31 | 2022-09-02 | 内蒙古工业大学 | 一种基于生成对抗网络的单幅图像去雾增强方法和系统 |
CN113450261A (zh) * | 2020-03-25 | 2021-09-28 | 江苏翼视智能科技有限公司 | 一种基于条件生成对抗网络的单幅图像去雾方法 |
CN111539885A (zh) * | 2020-04-21 | 2020-08-14 | 西安交通大学 | 一种基于多尺度网络的图像增强去雾方法 |
CN111539885B (zh) * | 2020-04-21 | 2023-09-19 | 西安交通大学 | 一种基于多尺度网络的图像增强去雾方法 |
CN111709888A (zh) * | 2020-06-03 | 2020-09-25 | 中科九度(北京)空间信息技术有限责任公司 | 一种基于改进的生成对抗网络的航拍图像去雾方法 |
CN111709888B (zh) * | 2020-06-03 | 2023-12-08 | 中科九度(北京)空间信息技术有限责任公司 | 一种基于改进的生成对抗网络的航拍图像去雾方法 |
CN111833306A (zh) * | 2020-06-12 | 2020-10-27 | 北京百度网讯科技有限公司 | 缺陷检测方法和用于缺陷检测的模型训练方法 |
CN111833306B (zh) * | 2020-06-12 | 2024-02-13 | 北京百度网讯科技有限公司 | 缺陷检测方法和用于缺陷检测的模型训练方法 |
CN111882495A (zh) * | 2020-07-05 | 2020-11-03 | 东北林业大学 | 一种基于自定义模糊逻辑与gan的图像高光处理方法 |
CN112308156B (zh) * | 2020-11-05 | 2022-05-03 | 电子科技大学 | 一种基于对抗学习的两阶段图像变化检测方法 |
CN112308156A (zh) * | 2020-11-05 | 2021-02-02 | 电子科技大学 | 一种基于对抗学习的两阶段图像变化检测方法 |
CN113554872A (zh) * | 2021-07-19 | 2021-10-26 | 昭通亮风台信息科技有限公司 | 一种位于交通路口及弯道的检测预警方法及系统 |
CN113554872B (zh) * | 2021-07-19 | 2022-09-20 | 昭通亮风台信息科技有限公司 | 一种位于交通路口及弯道的检测预警方法及系统 |
CN113724156A (zh) * | 2021-08-09 | 2021-11-30 | 浙江工业大学 | 一种结合大气散射模型的生成对抗网络去雾方法和系统 |
CN113724156B (zh) * | 2021-08-09 | 2024-03-29 | 浙江工业大学 | 一种结合大气散射模型的生成对抗网络去雾方法和系统 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109993804A (zh) | 一种基于条件生成对抗网络的道路场景去雾方法 | |
CN106910175B (zh) | 一种基于深度学习的单幅图像去雾算法 | |
CN111915530B (zh) | 一种基于端到端的雾霾浓度自适应神经网络图像去雾方法 | |
Chen et al. | Haze removal using radial basis function networks for visibility restoration applications | |
CN107292830B (zh) | 低照度图像增强及评价方法 | |
CN110544213A (zh) | 一种基于全局和局部特征融合的图像去雾方法 | |
CN111709888B (zh) | 一种基于改进的生成对抗网络的航拍图像去雾方法 | |
CN109816610A (zh) | 一种图像去雾系统 | |
CN109410144A (zh) | 一种基于深度学习的端到端图像去雾处理方法 | |
Bansal et al. | A review of image restoration based image defogging algorithms | |
CN116311254B (zh) | 一种恶劣天气情况下的图像目标检测方法、系统及设备 | |
CN114627269A (zh) | 一种基于深度学习目标检测的虚拟现实安防监控平台 | |
CN114429428A (zh) | 基于注意力机制的偏振图像去噪方法 | |
Qian et al. | CIASM-Net: a novel convolutional neural network for dehazing image | |
CN104318528A (zh) | 基于多尺度wls滤波的雾天图像恢复方法 | |
Cheng et al. | A highway traffic image enhancement algorithm based on improved GAN in complex weather conditions | |
CN111539896A (zh) | 一种基于域自适应的图像去雾方法及系统 | |
CN116452469B (zh) | 一种基于深度学习的图像去雾处理方法及装置 | |
CN116596792B (zh) | 一种面向智能船舶的内河雾天场景恢复方法、系统及设备 | |
CN110827375B (zh) | 一种基于微光图像的红外图像真彩着色方法及系统 | |
CN112164010A (zh) | 一种多尺度融合卷积神经网络图像去雾方法 | |
Zhou et al. | Multi-scale and attention residual network for single image dehazing | |
CN116310892A (zh) | 一种基于改进yolov4目标检测的海上救援方法 | |
CN112184566B (zh) | 一种针对附着水雾水珠去除的图像处理方法和系统 | |
CN111028184B (zh) | 一种图像增强方法及系统 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
RJ01 | Rejection of invention patent application after publication |
Application publication date: 20190709 |
|
RJ01 | Rejection of invention patent application after publication |