00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2010, Georgia Institute of Technology 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions 00009 * are met: 00010 * 00011 * * Redistributions of source code must retain the above copyright 00012 * notice, this list of conditions and the following disclaimer. 00013 * * Redistributions in binary form must reproduce the above 00014 * copyright notice, this list of conditions and the following 00015 * disclaimer in the documentation and/or other materials provided 00016 * with the distribution. 00017 * * Neither the name of the Georgia Institute of Technology nor the names of 00018 * its contributors may be used to endorse or promote products derived 00019 * from this software without specific prior written permission. 00020 * 00021 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00022 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00023 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 00024 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 00025 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 00026 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 00027 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00028 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 00029 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00030 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 00031 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00032 * POSSIBILITY OF SUCH DAMAGE. 00033 *********************************************************************/ 00034 00035 #ifndef gabor_filter_bank_h_DEFINED 00036 #define gabor_filter_bank_h_DEFINED 00037 00038 #include <opencv2/core/core.hpp> 00039 #include <opencv2/imgproc/imgproc.hpp> 00040 #include <opencv2/highgui/highgui.hpp> 00041 #include <vector> 00042 #include <iostream> 00043 00044 namespace cpl_visual_features 00045 { 00046 class GaborFilterBank 00047 { 00048 public: 00059 GaborFilterBank(int M=4, int N = 6, double U_l = 0.05, double U_h = 0.4, 00060 int gabor_size = 7); 00061 00069 std::vector<double> extractFeature(cv::Mat& frame); 00070 00080 double featureDist(std::vector<double> f1, std::vector<double> f2); 00081 00093 cv::Mat filterImg(cv::Mat& img, int m, int n, bool use_real=true); 00094 00095 protected: 00096 00101 void generateGaborFilters(); 00102 00106 void calcSampleVar(); 00107 00108 // 00109 // Class Members 00110 // 00111 00112 int M_; // Number of scale factors 00113 int N_; // Number of orientations 00114 double U_l_; // lower center frequency of interest 00115 double U_h_; // upper center frequency of interest 00116 int gabor_size_; // Pixel width of the gabor filter 00117 std::vector<std::vector<cv::Mat> > gabor_c_; // Complex gabor filters 00118 00119 // Vectors to store all extracted means and variances for calculation of the 00120 // Sample variances 00121 std::vector<std::vector<std::vector<double> > > alpha_base_mu_; 00122 std::vector<std::vector<std::vector<double> > > alpha_base_sigma_; 00123 // The vectors to store the smaple variances once calulcated 00124 std::vector<std::vector <double> > alpha_mu_; 00125 std::vector<std::vector <double> > alpha_sigma_; 00126 00127 bool calc_var_; // Trigger for needing to recalculate the sample variances 00128 00129 }; 00130 } 00131 00132 #endif // gabor_filter_bank_h_DEFINED