fun_mPb_theta.cpp
4.33 KB
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#include <stim/image/image.h>
#include <cmath>
//#include <conio.h>
#include <stim/visualization/colormap.h>
#include <stim/image/image_contour_detection.h>
void array_multiply(float* lhs, float rhs, unsigned int N);
void array_add(float* ptr1, float* ptr2, float* sum, unsigned int N);
/// This function evaluates the theta-dependent gradient image given an input image
/// @param lab is the 3-channel image in the LAB color space
/// @param theta is the angle used for computing the gradient
stim::image<float> func_mPb_theta(stim::image<float> lab, float theta, unsigned int w, unsigned int h){
//allocate space for the gradient image
stim::image<float> mPb_theta(w, h, 1);
//allocate space for each individual channel
stim::image<float> pic_light, pic_colora, pic_colorb;
pic_light = lab.channel(0);
pic_colora = lab.channel(1);
pic_colorb = lab.channel(2);
unsigned int N = w * h; //calculate the number of pixels in the image
float sigma = 2; //set the sigma value to \sigma = 2
unsigned int sigma_n = 3; //set the number of standard deviations used to define the kernel size
unsigned r1 = 3; //disk radii
unsigned r2 = 5;
unsigned r3 = 10;
unsigned r4 = 20;
float alpha[9] = {1,1,1,1,1,1,1,1,1}; //weighting for each channel
stim::image<float> l1,l2,l3,a1,a2,a3,b1,b2,b3;
l1 = gaussian_derivative_filter_odd(pic_light, sigma, sigma_n, r3 * 2, theta, w, h);
stim::cpu2image(l1.data(), "data_output/testl_tex5.bmp", w, h, stim::cmBrewer);
exit(0);
/*l2 = gaussian_derivative_filter_odd(pic_light, sigma, sigma_n, r2 * 2, theta, w, h);
stim::cpu2image(l2.data(), "data_output/l2_tex.bmp", w, h, stim::cmBrewer);
l3 = gaussian_derivative_filter_odd(pic_light, sigma, sigma_n, r3 * 2, theta, w, h);
stim::cpu2image(l3.data(), "data_output/l3_tex.bmp", w, h, stim::cmBrewer);
a1 = gaussian_derivative_filter_odd(pic_colora, sigma, sigma_n, r2 * 2, theta, w, h);
stim::cpu2image(a1.data(), "data_output/a1_tex.bmp", w, h, stim::cmBrewer);
a2 = gaussian_derivative_filter_odd(pic_colora, sigma, sigma_n, r3 * 2, theta, w, h);
stim::cpu2image(a2.data(), "data_output/a2_tex.bmp", w, h, stim::cmBrewer);
a3 = gaussian_derivative_filter_odd(pic_colora, sigma, sigma_n, r4 * 2, theta, w, h);
stim::cpu2image(a3.data(), "data_output/a3_tex.bmp", w, h, stim::cmBrewer);
b1 = gaussian_derivative_filter_odd(pic_colorb, sigma, sigma_n, r2 * 2, theta, w, h);
stim::cpu2image(b1.data(), "data_output/b1_tex.bmp", w, h, stim::cmBrewer);
b2 = gaussian_derivative_filter_odd(pic_colorb, sigma, sigma_n, r3 * 2, theta, w, h);
stim::cpu2image(b2.data(), "data_output/b2_tex.bmp", w, h, stim::cmBrewer);
b3 = gaussian_derivative_filter_odd(pic_colorb, sigma, sigma_n, r4 * 2, theta, w, h);
stim::cpu2image(b3.data(), "data_output/b3_tex.bmp", w, h, stim::cmBrewer);*/
/*for (unsigned i = 0; i<N; i++){
mPb_theta.data()[i] = l1.data()[i] * alpha[0] +
l2.data()[i] * alpha[1] +
l3.data()[i] * alpha[2] +
a1.data()[i] * alpha[3] +
a2.data()[i] * alpha[4] +
a3.data()[i] * alpha[5] +
b1.data()[i] * alpha[6] +
b2.data()[i] * alpha[7] +
b3.data()[i] * alpha[8] ;
}*/
array_multiply(l1.data(), alpha[0], N);
//stim::cpu2image(l1.data(), "data_output/array_add_l1.bmp", w, h, stim::cmBrewer);
array_multiply(l2.data(), alpha[1], N);
//stim::cpu2image(l2.data(), "data_output/array_add_l2.bmp", w, h, stim::cmBrewer);
array_multiply(l3.data(), alpha[2], N);
array_multiply(a1.data(), alpha[3], N);
array_multiply(a2.data(), alpha[4], N);
array_multiply(a3.data(), alpha[5], N);
array_multiply(b1.data(), alpha[6], N);
array_multiply(b2.data(), alpha[7], N);
array_multiply(b3.data(), alpha[8], N);
array_add(l1.data(), l2.data(), mPb_theta.data(), N);
//stim::cpu2image(sum, "data_output/array_add_sum.bmp", w, h, stim::cmBrewer);
array_add(mPb_theta.data(), l3.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), a1.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), a2.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), a3.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), b1.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), b2.data(), mPb_theta.data(), N);
array_add(mPb_theta.data(), b3.data(), mPb_theta.data(), N);
//stim::cpu2image(mPb_theta.data(), "data_output/mPb_theta0_1.bmp", w, h, stim::cmBrewer);
//getch();
return mPb_theta;
}