Peter M. Groen / openbr

Commit 2eaf99ee6ea6a65cb74d45cb6aec74b9c6bce7d3

Authored by Josh Klontz
1 parent a3fd6780

remove gabor

Inline Side-by-side
Showing 1 changed file with 0 additions and 206 deletions
openbr/plugins/imgproc/gabor.cpp deleted
View file @a3fd678
1 -/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *  
2 - * Copyright 2012 The MITRE Corporation *  
3 - * *  
4 - * Licensed under the Apache License, Version 2.0 (the "License"); *  
5 - * you may not use this file except in compliance with the License. *  
6 - * You may obtain a copy of the License at *  
7 - * *  
8 - * http://www.apache.org/licenses/LICENSE-2.0 *  
9 - * *  
10 - * Unless required by applicable law or agreed to in writing, software *  
11 - * distributed under the License is distributed on an "AS IS" BASIS, *  
12 - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *  
13 - * See the License for the specific language governing permissions and *  
14 - * limitations under the License. *  
15 - * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */  
16 -  
17 -#include <opencv2/imgproc/imgproc.hpp>  
18 -  
19 -#include <openbr/plugins/openbr_internal.h>  
20 -  
21 -using namespace cv;  
22 -  
23 -namespace br  
24 -{  
25 -  
26 -/*!  
27 - * \ingroup transforms  
28 - * \brief Implements a Gabor Filter  
29 - * \br_link http://en.wikipedia.org/wiki/Gabor_filter  
30 - * \author Josh Klontz \cite jklontz  
31 - */  
32 -class GaborTransform : public UntrainableTransform  
33 -{  
34 - Q_OBJECT  
35 - Q_ENUMS(Component)  
36 - Q_PROPERTY(float lambda READ get_lambda WRITE set_lambda RESET reset_lambda STORED false)  
37 - Q_PROPERTY(float theta READ get_theta WRITE set_theta RESET reset_theta STORED false)  
38 - Q_PROPERTY(float psi READ get_psi WRITE set_psi RESET reset_psi STORED false)  
39 - Q_PROPERTY(float sigma READ get_sigma WRITE set_sigma RESET reset_sigma STORED false)  
40 - Q_PROPERTY(float gamma READ get_gamma WRITE set_gamma RESET reset_gamma STORED false)  
41 - Q_PROPERTY(Component component READ get_component WRITE set_component RESET reset_component STORED false)  
42 -  
43 -public:  
44 - /*!< */  
45 - enum Component { Real,  
46 - Imaginary,  
47 - Magnitude,  
48 - Phase };  
49 -  
50 -private:  
51 - BR_PROPERTY(float, lambda, 0)  
52 - BR_PROPERTY(float, theta, 0)  
53 - BR_PROPERTY(float, psi, 0)  
54 - BR_PROPERTY(float, sigma, 0)  
55 - BR_PROPERTY(float, gamma, 0)  
56 - BR_PROPERTY(Component, component, Phase)  
57 -  
58 - Mat kReal, kImaginary;  
59 -  
60 - friend class GaborJetTransform;  
61 -  
62 - static void makeWavelet(float lambda, float theta, float psi, float sigma, float gamma, Mat &kReal, Mat &kImaginary)  
63 - {  
64 - float sigma_x = sigma;  
65 - float sigma_y = sigma/gamma;  
66 -  
67 - // Bounding box  
68 - const double nstds = 3;  
69 - int xmax = std::ceil(std::max(1.0, std::max(std::abs(nstds*sigma_x*cos(theta)),  
70 - std::abs(nstds*sigma_y*sin(theta)))));  
71 - int ymax = std::ceil(std::max(1.0, std::max(std::abs(nstds*sigma_x*sin(theta)),  
72 - std::abs(nstds*sigma_y*cos(theta)))));  
73 -  
74 - // Compute kernels  
75 - kReal.create(2*ymax+1, 2*xmax+1, CV_32FC1);  
76 - kImaginary.create(2*ymax+1, 2*xmax+1, CV_32FC1);  
77 - for (int y = -ymax; y <= ymax; y++) {  
78 - int row = y + ymax;  
79 - for (int x = -xmax; x <= xmax; x++) {  
80 - int col = x + xmax;  
81 - float x_prime = x*cos(theta) + y*sin(theta);  
82 - float y_prime = -x*sin(theta) + y*cos(theta);  
83 - float a = exp(-0.5 * (x_prime*x_prime + gamma*gamma*y_prime*y_prime)/(sigma*sigma));  
84 - float b = 2*CV_PI*x_prime/lambda+psi;  
85 - kReal.at<float>(row, col) = a*cos(b);  
86 - kImaginary.at<float>(row, col) = a*sin(b);  
87 - }  
88 - }  
89 -  
90 - // Remove DC component, should only effect real kernel  
91 - subtract(kReal, mean(kReal), kReal);  
92 - subtract(kImaginary, mean(kImaginary), kImaginary);  
93 - }  
94 -  
95 - void init()  
96 - {  
97 - makeWavelet(lambda, theta, psi, sigma, gamma, kReal, kImaginary);  
98 - }  
99 -  
100 - void project(const Template &src, Template &dst) const  
101 - {  
102 - Mat real, imaginary, magnitude, phase;  
103 - if (component != Imaginary)  
104 - filter2D(src, real, -1, kReal);  
105 - if (component != Real)  
106 - filter2D(src, imaginary, -1, kImaginary);  
107 - if ((component == Magnitude) || (component == Phase))  
108 - cartToPolar(real, imaginary, magnitude, phase);  
109 -  
110 - if (component == Real) dst = real;  
111 - else if (component == Imaginary) dst = imaginary;  
112 - else if (component == Magnitude) dst = magnitude;  
113 - else if (component == Phase) dst = phase;  
114 - else qFatal("Invalid component.");  
115 - }  
116 -};  
117 -  
118 -BR_REGISTER(Transform, GaborTransform)  
119 -  
120 -/*!  
121 - * \ingroup transforms  
122 - * \brief A vector of gabor wavelets applied at a point.  
123 - * \author Josh Klontz \cite jklontz  
124 - */  
125 -class GaborJetTransform : public UntrainableTransform  
126 -{  
127 - Q_OBJECT  
128 - Q_ENUMS(br::GaborTransform::Component)  
129 - Q_PROPERTY(QList<float> lambdas READ get_lambdas WRITE set_lambdas RESET reset_lambdas STORED false)  
130 - Q_PROPERTY(QList<float> thetas READ get_thetas WRITE set_thetas RESET reset_thetas STORED false)  
131 - Q_PROPERTY(QList<float> psis READ get_psis WRITE set_psis RESET reset_psis STORED false)  
132 - Q_PROPERTY(QList<float> sigmas READ get_sigmas WRITE set_sigmas RESET reset_sigmas STORED false)  
133 - Q_PROPERTY(QList<float> gammas READ get_gammas WRITE set_gammas RESET reset_gammas STORED false)  
134 - Q_PROPERTY(br::GaborTransform::Component component READ get_component WRITE set_component RESET reset_component STORED false)  
135 - BR_PROPERTY(QList<float>, lambdas, QList<float>())  
136 - BR_PROPERTY(QList<float>, thetas, QList<float>())  
137 - BR_PROPERTY(QList<float>, psis, QList<float>())  
138 - BR_PROPERTY(QList<float>, sigmas, QList<float>())  
139 - BR_PROPERTY(QList<float>, gammas, QList<float>())  
140 - BR_PROPERTY(GaborTransform::Component, component, GaborTransform::Phase)  
141 -  
142 - QList<Mat> kReals, kImaginaries;  
143 -  
144 - void init()  
145 - {  
146 - kReals.clear();  
147 - kImaginaries.clear();  
148 - foreach (float lambda, lambdas)  
149 - foreach (float theta, thetas)  
150 - foreach (float psi, psis)  
151 - foreach (float sigma, sigmas)  
152 - foreach (float gamma, gammas) {  
153 - Mat kReal, kImaginary;  
154 - GaborTransform::makeWavelet(lambda, theta, psi, sigma, gamma, kReal, kImaginary);  
155 - kReals.append(kReal);  
156 - kImaginaries.append(kImaginary);  
157 - }  
158 - }  
159 -  
160 - static float response(const cv::Mat &src, const QPointF &point, const Mat &kReal, const Mat &kImaginary, GaborTransform::Component component)  
161 - {  
162 - Rect roi(std::max(std::min((int)(point.x() - kReal.cols/2.f), src.cols - kReal.cols), 0),  
163 - std::max(std::min((int)(point.y() - kReal.rows/2.f), src.rows - kReal.rows), 0),  
164 - kReal.cols,  
165 - kReal.rows);  
166 -  
167 - float real = 0, imaginary = 0, magnitude = 0, phase = 0;  
168 - if (component != GaborTransform::Imaginary) {  
169 - Mat dst;  
170 - multiply(src(roi), kReal, dst);  
171 - real = sum(dst)[0];  
172 - }  
173 - if (component != GaborTransform::Real) {  
174 - Mat dst;  
175 - multiply(src(roi), kImaginary, dst);  
176 - imaginary = sum(dst)[0];  
177 - }  
178 - if ((component == GaborTransform::Magnitude) || (component == GaborTransform::Phase)) {  
179 - magnitude = sqrt(real*real + imaginary*imaginary);  
180 - phase = atan2(imaginary, real)*180/CV_PI;  
181 - }  
182 -  
183 - float dst = 0;  
184 - if (component == GaborTransform::Real) dst = real;  
185 - else if (component == GaborTransform::Imaginary) dst = imaginary;  
186 - else if (component == GaborTransform::Magnitude) dst = magnitude;  
187 - else if (component == GaborTransform::Phase) dst = phase;  
188 - else qFatal("Invalid component.");  
189 - return dst;  
190 - }  
191 -  
192 - void project(const Template &src, Template &dst) const  
193 - {  
194 - const QList<QPointF> points = src.file.points();  
195 - dst = Mat(points.size(), kReals.size(), CV_32FC1);  
196 - for (int i=0; i<points.size(); i++)  
197 - for (int j=0; j<kReals.size(); j++)  
198 - dst.m().at<float>(i,j) = response(src, points[i], kReals[j], kImaginaries[j], component);  
199 - }  
200 -};  
201 -  
202 -BR_REGISTER(Transform, GaborJetTransform)  
203 -  
204 -} // namespace br  
205 -  
206 -#include "imgproc/gabor.moc"