1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#include "dcp_core.h"
using namespace cv::ximgproc;
void dehaze(IplImage *recover, IplImage *input)
{
int height = input->height;
int width = input->width;
IplImage *darkchannel = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
IplImage *transmission = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
IplImage *refine_transmission = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
int darkchannelradius = cvRound(MIN(width, height) * 0.02);
double Airlight[3] = { 0.0, 0.0, 0.0 };
TicToc t;
t.tic();
printf("CalcDarkChannel...");
CalcDarkChannel(darkchannel, input, darkchannelradius);
t.toc();
t.tic();
printf("CalcAirlight...");
CalcAirlight(darkchannel, input, Airlight);
t.toc();
t.tic();
printf("CalcTransmission...");
CalcTransmission(transmission, input, Airlight, darkchannelradius);
t.toc();
t.tic();
printf("GuidedFilterColor...");
// GuidedFilterColor() is my own implementation of guided filter
// guidedFilter() is the implementation included in OpenCV 3.0
// The result is almost the same, however my implementation is much more slower
// because I haven't spend much time optimizing the code efficiency.
// See http://research.microsoft.com/en-us/um/people/kahe/eccv10/ for more details
// ref:
// He, Kaiming, Jian Sun, and Xiaoou Tang. "Guided image filtering."
// Pattern Analysis and Machine Intelligence, IEEE Transactions on 35.6 (2013): 1397-1409.
GuidedFilterColor(refine_transmission, input, transmission, 1e-6, 60);
// guidedFilter(cv::cvarrToMat(input), cv::cvarrToMat(transmission), cv::cvarrToMat(refine_transmission), 60, 1e-6);
t.toc();
t.tic();
printf("CalcRecover...");
CalcRecover(recover, input, refine_transmission, Airlight);
t.toc();
cvReleaseImage(&refine_transmission);
cvReleaseImage(&transmission);
cvReleaseImage(&darkchannel);
}
|
