Single Image Haze Removal(图像去雾)-CVPR’09 Best Paper

【原文:http://blog.csdn.net/abcjennifer/article/details/6662706

真正了解了什么叫最简单的就是最美好的

真正的好文章不需要大堆公式堆积显得充实,而是最最平实的思想!

这篇文章的:原文PDFPPT。感兴趣的可以了解一下。

这篇文章的目的就是以最简单的思路将图像达到去雾效果。用Matlab编了一下,效果图特别好啊哈:




下面是摘录的:

CVPR的中文名是计算机视觉与模式识别会议,是计算机视觉领域最顶尖的国际 会议之一。09年的CVPR共收到约1450篇投稿,其中393篇文章被接收,接收率为26%。只有一篇文章被选为今年的最佳论文。这是CVPR创立25年 以来首次由中国人获得这个奖项。

2010年的结果也已经出来了,一共1724篇文章,CVPR2010 Paper Acceptance Rates: 78 papers were accepted for ORAL Presentation (4.5%).384 papers were accepted for POSTER Presentation (22.3%).

下面是作者的感想摘录,值得借鉴:

这篇论文研究的问题是图像的去雾技术,它可 以还原图像的颜色和能见度,同时也能利用雾的浓度来估计物体的距离,这些在计算机视觉上都有重要应用(例如三维重建,物体识别)。但是之前人们还没找到简 单有效的方法来达到这个目的。在这篇论文里,我们找到了一个非常简单的,甚至说令人惊讶统计规律,并提出了有效的去雾方法。

与之前的方法不同,我们把注意力放到了无雾图像的统计特征上。我们发 现,在无雾图像中,每一个局部区域都很有可能会有阴影,或者是纯颜色的东西,又或者是黑色的东西。因此,每一个局部区域都很有可能有至少一个颜色通道会有 很低的值。我们把这个统计规律叫做Dark Channel Prior。直观来说,Dark Channel Prior认为每一个局部区域都总有一些很暗的东西。这个规律很简单,但在我们研究的去雾问题上却是本质的基本规律。

由于雾总是灰白色的,因此一旦图像受到雾的影响,那么这些本来应该很暗 的东西就会变得灰白。不仅如此,根据物理上雾的形成公式,我们还能根据这些东西的灰白程度来判断雾的浓度。因此,我们提出的Dark Channel Prior能很有效地去除雾的影响,同时利用物的浓度来估算物体的距离。

大道之行在于简

我们这篇文章的三个审稿人都给出了最高的评分。他们认为我们的方法简单 而有效。其中一位评委说,Dark Channel Prior的想法听起来很不可思议,但我们却证明了其真实性。另一位评委认为很少有文章能够用如此简单的方法使实验结果获得如此大的提升。还有一位评委甚 至亲自实现了我们的方法并确认其可行。孙剑说阅读这样的评审结果是一件让人快乐的事情。而汤老师认为,这篇文章的成功在于三个方面。第一,方法非常简单; 第二,对于一个很困难的问题,给出了很好的结果;第三,发现了一个基本的自然规律并且应用在实际的问题中。在迈阿密的演讲结束后,观众也给予了很高的评 价。他们跟我说,这是这次CVPR上最有趣的一个演讲。

一位与会的研究员说,最好的idea,往往就是那些看起来很简单,但说 出来大家都会觉得怎么没有人想到过的idea。而我们的idea正好就符合了这一点。我们论文摘要的第一句话是这么说的,“我们提出了一个简单而有效的方 法”。或许,这就是对我们这次工作最好的概括——简单的,就是美的。


代码如下:

1.MATLAB

[html]  view plain  copy
  1. %=========================================================%  
  2. %调用规则:(有雾时调用,否则不调用)  
  3. %实际操作时,according to experiments:  
  4. %percent=under_50/total  
  5. %percent<0.1%,取w=0.6  
  6. %percent>0.1%&&percent<1%,取w=0.45  
  7. %percenet>1%&&percent<2%,取w=0.3  
  8. %else not use haze-free-adjust  
  9. %有雾:绘制出的直方图<50的部分<1%  
  10. %最后控制台还会输出原图中under_50像素点所占比例  
  11. %=========================================================%  
  12. close all  
  13. clear all  
  14. clc  
  15. blockSize=15;               %每个block为15个像素  
  16. w0=0.6;                      
  17. t0=0.1;  
  18. A=200;  
  19. I=imread('D:\毕业设计\Images\pic_loc\1870575550604291415.jpg');  
  20. %I=imread('C:\Users\Zrq\Desktop\同济.jpg');  
  21. h = figure;  
  22. %set(gcf,'outerposition',get(0,'screensize'));%获得SystemScreenSize 传递给当前图像句柄gcf的outerposition属性  
  23. subplot(321)%表示3(行数)*2(列数)的图像,1代表所画图形的序号  
  24. imshow(I);  
  25. title('Original Image');  
  26.   
  27.   
  28. subplot(323);  
  29. grayI=rgb2gray(I);  
  30. imshow(grayI,[]);  
  31. title('原图像灰度图')  
  32.   
  33. subplot(324);  
  34. imhist(grayI,64);  
  35.   
  36. %统计<50的像素所占的比例  
  37. %%%%%%%%%%%%%%%%%%%%%%  
  38. [COUNT x]=imhist(grayI);  
  39. under_50=0;  
  40. for i=0:50  
  41.     under_50=under_50+COUNT(x==i);  
  42. end  
  43. under_50  
  44. total=size(I,1)*size(I,2)*size(I,3);  
  45. percent=under_50/total  
  46. %%%%%%%%%%%%%%%%%%%%%%  
  47.   
  48. if(percent>0.02)  
  49.     error('This image need not Haze-Free-Proprocessing.');  
  50. else if(percent<0.001)  
  51.         w=0.6;  
  52.     else if (percent>0.01)  
  53.             w=0.3;  
  54.         else  
  55.             w=0.45;  
  56.         end  
  57.     end  
  58. end  
  59.   
  60. [h,w,s]=size(I);  
  61. min_I=zeros(h,w);  
  62.   
  63. for i=1:h                   
  64.     for j=1:w  
  65.         dark_I(i,j)=min(I(i,j,:));%取每个点的像素为RGB分量中最低的那个通道的值  
  66.     end  
  67. end  
  68.   
  69. Max_dark_channel=double(max(max(dark_I)))  
  70. dark_channel=double(dark_I);  
  71. t=1-w0*(dark_channel/Max_dark_channel);  
  72.   
  73. T=uint8(t*255);  
  74.   
  75. t=max(t,t0);  
  76.   
  77. %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  
  78. I1=double(I);  
  79. J(:,:,1) = uint8((I1(:,:,1) - (1-t)*Max_dark_channel)./t);  
  80.   
  81. J(:,:,2) = uint8((I1(:,:,2) - (1-t)*Max_dark_channel)./t);  
  82.   
  83. J(:,:,3) =uint8((I1(:,:,3) - (1-t)*Max_dark_channel)./t);  
  84. subplot(322)  
  85. imshow(J);  
  86. imwrite(J,'tj2.jpg');  
  87. title('Haze-Free Image:');  
  88.   
  89. subplot(325);  
  90. grayJ=rgb2gray(J);  
  91. imshow(grayJ,[]);  
  92. title('去雾后灰度图')  
  93.   
  94. subplot(326);  
  95. imhist(grayJ,64);  

2.OPENCV

[cpp]  view plain  copy
  1. //Hazefree helper  
  2.   
  3. char tbarname1[] = "调节block";  
  4. //定义两个滑动条,用于调节参数  
  5. char tbarname2[] = "调节w";  
  6. //w是为了保留一部分的雾  
  7. int block=5;  
  8. int w1=80;  
  9. double w;  
  10. IplImage *dst=NULL;  
  11.   
  12. IplImage *quw(IplImage *src,int block,double w)  
  13. {  
  14.     //图像分别有三个颜色通道  
  15.     IplImage *dst1=NULL;  
  16.     IplImage *dst2=NULL;  
  17.     IplImage *dst3=NULL;  
  18.     IplImage *imgroi1;  
  19.     //dst1的ROI  
  20.     IplImage *imgroi2;  
  21.     //dst2的ROI  
  22.     IplImage *imgroi3;  
  23.     //dst3的ROI  
  24.     IplImage *roidark;  
  25.     //dark channel的ROI  
  26.     IplImage *dark_channel=NULL;  
  27.     //暗原色先验的指针  
  28.     IplImage *toushelv=NULL;  
  29.     //透射率  
  30.   
  31.     //去雾算法运算后的三个通道  
  32.     IplImage *j1=NULL;  
  33.     IplImage *j2=NULL;  
  34.     IplImage *j3=NULL;  
  35.     //去雾后的图像,三通道合并成  
  36.     IplImage *dst=NULL;  
  37.     //源图像ROI位置以及大小  
  38.     CvRect ROI_rect;  
  39.   
  40.     //分离的三个通道  
  41.     dst1=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  42.     dst2=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  43.     dst3=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  44.   
  45.     //为各个ROI分配内存  
  46.     imgroi1=cvCreateImage(cvSize(block,block),IPL_DEPTH_8U,1);  
  47.     imgroi2=cvCreateImage(cvSize(block,block),IPL_DEPTH_8U,1);  
  48.     imgroi3=cvCreateImage(cvSize(block,block),IPL_DEPTH_8U,1);  
  49.     roidark=cvCreateImage(cvSize(block,block),IPL_DEPTH_8U,1);  
  50.   
  51.     //为j1 j2 j3分配大小  
  52.     j1=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  53.     j2=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  54.     j3=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  55.   
  56.     //为暗原色先验指针分配大小  
  57.     dark_channel=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  58.     //为透射率指针分配大小  
  59.     toushelv=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,1);  
  60.     //dst分配大小  
  61.     dst=cvCreateImage(cvSize(src->width,src->height),IPL_DEPTH_8U,3);  
  62.     //将原彩色图像分离成三通道  
  63.     cvSplit(src,dst1,dst2,dst3,NULL);  
  64.     //求暗原色  
  65.     ROI_rect.width=block;  
  66.     ROI_rect.height=block;  
  67.     ROI_rect.x=0;  
  68.     ROI_rect.y=0;  
  69.   
  70.   
  71.     int i;  
  72.     int j;  
  73.     double min1=0;  
  74.     double max1=0;  
  75.     double min2=0;  
  76.     double max2=0;  
  77.     double min3=0;  
  78.     double max3=0;  
  79.     double min=0;  
  80.     CvScalar value;  
  81.     for(i=0;i<src->width/block;i++)  
  82.     {        for(j=0;j<src->height/block;j++)  
  83.     {  
  84.         //分别计算三个通道内ROI的最小值  
  85.         cvSetImageROI(dst1,ROI_rect);  
  86.         cvCopy(dst1,imgroi1,NULL);  
  87.         cvMinMaxLoc(imgroi1,&min1,&max1,NULL,NULL);  
  88.         cvSetImageROI(dst2,ROI_rect);  
  89.         cvCopy(dst2,imgroi2,NULL);  
  90.         cvMinMaxLoc(imgroi2,&min2,&max2,NULL,NULL);  
  91.         cvSetImageROI(dst3,ROI_rect);  
  92.         cvCopy(dst3,imgroi3,NULL);  
  93.         cvMinMaxLoc(imgroi3,&min3,&max3,NULL,NULL);  
  94.         //求三个通道内最小值的最小值  
  95.         if(min1<min2)  
  96.             min=min1;  
  97.         else  
  98.             min=min2;  
  99.         if(min>min3)  
  100.             min=min3;//min为这个ROI中暗原色  
  101.         value=cvScalar(min,min,min,min);//min放在value中  
  102.         //min赋予dark_channel中相应的ROI  
  103.         cvSetImageROI(dark_channel,ROI_rect);  
  104.         cvSet(roidark,value,NULL);  
  105.         cvCopy(roidark,dark_channel,NULL);  
  106.         //释放各个ROI  
  107.         cvResetImageROI(dst1);  
  108.         cvResetImageROI(dst2);  
  109.         cvResetImageROI(dst3);  
  110.         cvResetImageROI(dark_channel);  
  111.         //转入下一个ROI  
  112.         ROI_rect.x=block*i;  
  113.         ROI_rect.y=block*j;  
  114.     }  
  115.     }  
  116.     //保存暗原色先验的图像  
  117.     cvSaveImage("f:/dark_channel_prior.jpg",dark_channel);  
  118.     //利用得到的暗原色先验dark_channel_prior.jpg求大气光强  
  119.     double min_dark;  
  120.     double max_dark;  
  121.     CvPoint min_loc;  
  122.     CvPoint max_loc;//max_loc是暗原色先验最亮一小块的原坐标  
  123.     cvMinMaxLoc(dark_channel,&min_dark,&max_dark,&min_loc,&max_loc,NULL);  
  124. //  cout<<max_loc.x<<" "<<max_loc.y<<endl;  
  125.     ROI_rect.x=max_loc.x;  
  126.     ROI_rect.y=max_loc.y;  
  127.     double A_dst1;//定义大气光成分的估计值  
  128.     double dst1_min;  
  129.     double A_dst2;  
  130.     double dst2_min;  
  131.     double A_dst3;  
  132.     double dst3_min;  
  133.     cvSetImageROI(dst1,ROI_rect);  
  134.     //按照论文方法求大气光强估计值  
  135.     cvCopy(dst1,imgroi1,NULL);  
  136.     cvMinMaxLoc(imgroi1,&dst1_min,&A_dst1,NULL,NULL);  
  137.     cvSetImageROI(dst2,ROI_rect);  
  138.     cvCopy(dst2,imgroi2,NULL);  
  139.     cvMinMaxLoc(imgroi2,&dst2_min,&A_dst2,NULL,NULL);  
  140.     cvSetImageROI(dst3,ROI_rect);  
  141.     cvCopy(dst3,imgroi3,NULL);  
  142.     cvMinMaxLoc(imgroi3,&dst3_min,&A_dst3,NULL,NULL);  
  143. //  cout<<A_dst1<<" "<<A_dst2<<" "<<A_dst3<<endl;//这三值为大气光强度估计值  
  144.     //求透射率  
  145.     int k;  
  146.     int l;  
  147.     CvScalar m;  
  148.     CvScalar n;//暗原色先验各元素值  
  149.   
  150.     for(k=0;k<src->height;k++)  
  151.     {  
  152.         for(l=0;l<src->width;l++)  
  153.         {  
  154.             m=cvGet2D(dark_channel,k,l);  
  155.             n=cvScalar(255-w*m.val[0]);  
  156.             //w目的是保留一部分的雾,使图像看起来真实些  
  157.             cvSet2D(toushelv,k,l,n);  
  158.         }  
  159.     }  
  160.     cvSaveImage("f:/toushelv.jpg",toushelv);  
  161.   
  162.     //求无雾图像  
  163.     int p,q;  
  164.     double tx;  
  165.     double jj1,jj2,jj3;  
  166.     CvScalar ix,jx;  
  167.     for(p=0;p<src->height;p++)  
  168.     {  
  169.         for(q=0;q<src->width;q++)  
  170.         {  
  171.             tx=cvGetReal2D(toushelv,p,q);  
  172.             tx=tx/255;  
  173.             if(tx<0.1)  
  174.                 tx=0.1;  
  175.             ix=cvGet2D(src,p,q);  
  176.             jj1=(ix.val[0]-A_dst1)/tx+A_dst1;//根据雾产生模型运算,还原出无雾图像  
  177.             jj2=(ix.val[1]-A_dst2)/tx+A_dst2;  
  178.             jj3=(ix.val[2]-A_dst3)/tx+A_dst3;  
  179.             jx=cvScalar(jj1,jj2,jj3,0.0);  
  180.             cvSet2D(dst,p,q,jx);  
  181.         }  
  182.     }  
  183.     cvSaveImage("f:/removed_haze.jpg",dst);  
  184.   
  185.     //释放指针  
  186.     cvReleaseImage(&dst1);  
  187.     cvReleaseImage(&dst2);  
  188.     cvReleaseImage(&dst3);  
  189.     cvReleaseImage(&imgroi1);  
  190.     cvReleaseImage(&imgroi2);  
  191.     cvReleaseImage(&imgroi3);  
  192.     cvReleaseImage(&roidark);  
  193.     cvReleaseImage(&dark_channel);  
  194.     cvReleaseImage(&toushelv);  
  195.     cvReleaseImage(&j1);  
  196.     cvReleaseImage(&j2);  
  197.     cvReleaseImage(&j3);  
  198.     return dst;  
  199. }  
  200.   
  201. IplImage *source;  
  202. void on_trackbar1(int h)  
  203. {  
  204.     dst=quw(source,block,w);  
  205.     cvShowImage("目的图像",dst);  
  206.     //      cvWaitKey(0);  
  207. }  
  208. void on_trackbar2(int h)  
  209. {  
  210.     w=(double)w1/100;  
  211.     dst=quw(source,block,w);  
  212.     cvShowImage("目的图像",dst);  
  213.     //      cvWaitKey(0);  
  214. }  
  215.   
  216. void CCVMFCView::OnImageHazefree()  
  217. {  
  218.     imageClone(workImg,&source);  
  219.     //cvNamedWindow("有雾图像",CV_WINDOW_AUTOSIZE);  
  220.     cvFlip(source);  
  221.     //cvShowImage("有雾图像",source);  
  222.     cvNamedWindow("目的图像",CV_WINDOW_AUTOSIZE);  
  223.     cvShowImage("目的图像",source);  
  224.     cvCreateTrackbar(tbarname1, "目的图像", &block, 15, on_trackbar1);  
  225.     cvCreateTrackbar(tbarname2, "目的图像", &w1, 100, on_trackbar2);  
  226.     cvWaitKey(0);  
  227.       
  228.     cvReleaseImage(&source);  
  229.   
  230.     cvDestroyWindow("目的图像");  
  231.     cvDestroyWindow("有雾图像");  
  232.     cvFlip(dst);  
  233.     m_dibFlag=imageReplace(dst,&workImg);  
  234.     Invalidate();  
  235. }  

猜你喜欢

转载自blog.csdn.net/zhazhiqiang2010/article/details/79103611