本发明属于图像增强领域，具体涉及一种基于单幅图像的改进暗通道去雾方法。
背景技术：
在雾天条件下，光线在传播过程中会收到空气中各种介质的散射，导致可视能力大大降低。这一情况会导致马路交通拥堵、航班行程耽误，甚至于一些军事行动受到大幅度的影响。因此雾天甚至于大雾天气对我们的生活带来了诸多不便。为了改善这种问题的影响，针对去雾算法的研究势在必行。
目前去雾算法主要分为不依赖模型和依赖模型两类。不依赖模型采用的是传统的类似于对比度拉伸、直方图均衡化、伽马矫正等图像增强算法。依赖模型采用的有几种方法，narasimhan等根据改进的大气散射模型进行去雾恢复；yoavy.schechner通过依赖偏振片来实现自动分离去雾方法；还有一种经典算法就是he在2009年推出的基于暗原色先验的单幅图像去雾算法。
上述这些算法对一定条件下的雾天场景有去雾效果，其中最有效的就是he所提出的基于暗原色先验的单幅图像去雾算法，但是都在实际场景应用中存在实时性不足的缺点。
技术实现要素：
本发明要解决技术问题为：针对上述传统算法实时性不足的特点，本发明基于暗原色先验的单幅图像去雾算法，添加对边缘部分的处理，并且为了消除边缘阴影区的影响，加入形态学膨胀腐蚀操作。目的在于快速提升去雾算法的处理时间。
本发明采用的技术方案为：基于单幅图像的改进暗通道去雾方法，具体包含如下步骤：
步骤1：输入图像为彩色图像i，对彩色图像的r通道分量i(,,1)利用canny算子求取图像内边缘特性，得到二值化边缘特性图像i_edge；
步骤2：针对边缘特性图像i_edge，进行一次形态学膨胀操作，得到膨胀后边缘特性图像i_pz_edge；
步骤3：对非边缘位置求取暗通道图像jdark(i,j)，并对边缘位置作保留处理，
其中，i,j表示图像的二维坐标；jdark(i,j)为要求取的暗通道图像；i_pz_edge(i,j)为膨胀后的边缘图像，其为0的时候表示的为非边缘图像部分；ic为原图，mincic为三通道分量中最小的一个分量值；y∈ω(x)表示局部窗口内；
步骤4：求取全球大气光值a，对暗通道图像jdark(i，j)作简易直方图统计，求取灰度值大于220的像素点的平均灰度值，以作为全球大气光值a；
步骤5：求取粗透射率图t1
t1＝1-w*jdark
其中，t1为粗透射率图；jdark为暗通道图像；w为引入的保留一定雾气的常数；
步骤6：求取精透射率图t，对粗透射率图t1进行形态学腐蚀操作，获得更为精确的透射率图t；
步骤7：利用输入图、精透射率图和全球大气光参数，求取去雾后图像jbright，
其中，canny算子所采用的高低阈值比例为5:2。
其中，形态学膨胀所采用的窗口大小为5。
其中，求取图像的暗通道时，对比膨胀后的边缘图像的边缘位置，对应位置作保留原像素值处理，其它位置求取邻域内极小值，邻域窗口大小为15。
其中，在求取全球大气光值的时候，对灰度220以上的像素点作简易直方图统计然后求取均值作为全球大气光值a。
其中，粗透射率图保留一定程度的雾气成分，采用的保留参数w值为0.96。
其中，求取精透射率图时所采用的形态学腐蚀操作，窗口大小为5。
本发明与现有技术相比，其显著优点：(1)首先都原始输入图像作边缘检测，并针对边缘检测区域单薄容易带来阴影区的特点，采用膨胀操作适量扩大边缘区域，提升暗通道图像的准确率(2)在求取全球大气光值时，抛开原先的按像素比例划分区域，直接求取给定像素阈值之上的灰度均值，大幅度缩减计算时间。(3)针对粗透射率图进行一次腐蚀操作，能够有效提升透射率图的准确性，降低去雾效果图中的边缘虚化效应。
附图说明
图1为本发明基于单幅图像的改进暗通道去雾方法工作流程示意图；
图2为输入有雾图像示意图；
图3为边缘提取示意图；
图4为暗通道图像；
图5为粗透射率图；
图6为精透射率图；
图7为去雾后效果图。
具体实施方式
下面结合附图以及具体实施方式进一步说明本发明。
结合图1，本发明基于单幅图像的改进暗通道去雾方法，具体步骤如下：
步骤1：如图2和图3所示，对输入图像进行canny算子边缘提取，并对边缘提取后的图像进行一级膨胀操作，膨胀窗口为5*5。求得边缘图像i_pz_edge。
步骤2：如图4所示，求取暗通道图像。具体根据图3得到的边缘区域位置的像素点，对输入图像其它位置的像素点均求取领域内的极小值并替换当前位置像素值，得到全局暗通道图像jdark(i,j)。
jdark(i,j)为要求取的暗通道图像，其中i,j为图像二维坐标；i_pz_edge(i,j)为膨胀后的边缘图像，其为0的时候表示的为非边缘图像部分；ic为原图，mincic为三通道分量中最小的一个分量值；y∈ω(x)表示局部窗口内。
步骤3：求取大气光值a，针对图4暗通道图像，求取灰度值大于220的像素点的灰度均值作为大气光值a。
步骤4：求取粗透射率图。利用求得的暗通道图像jdark，并增添雾气保留参数w，取值为0.96，计算得到粗透射率图t1。如图5所示。
t1＝1-w*jdark
其中，t1为粗透射率图；jdark为暗通道图像；w为引入的保留一定雾气的常数；
步骤5：求取精透射率图t。针对图5进行一级腐蚀操作，求得精透射率图t。
步骤6：恢复得到去雾后效果图。如图6所示。即输入图、精透射率图和全球大气光参数，求取去雾后图像jbright。
本发明未详细阐述部分属于本领域公知技术。