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 color_histogram_h_DEFINED 00036 #define color_histogram_h_DEFINED 00037 00038 #include <opencv2/core/core.hpp> 00039 #include <opencv2/imgproc/imgproc.hpp> 00040 #include "abstract_feature.h" 00041 namespace cpl_visual_features 00042 { 00043 template<int n_bins> class ColorHistogram : public AbstractFeature<cv::MatND> 00044 { 00045 public: 00046 ColorHistogram() 00047 { 00048 } 00049 00050 virtual void operator()(cv::Mat& patch, cv::Rect& window) 00051 { 00052 int bins[] = {n_bins, n_bins, n_bins}; 00053 int channels[] = {0,1,2}; 00054 float r_ranges[] = {0,256}; 00055 float g_ranges[] = {0,256}; 00056 float b_ranges[] = {0,256}; 00057 const float* ranges[] = {r_ranges, g_ranges, b_ranges}; 00058 00059 cv::MatND patch_desc; 00060 00061 cv::calcHist(&patch, 1, channels, cv::Mat(), patch_desc, 3, bins, ranges, 00062 true, false); 00063 descriptor_ = patch_desc; 00064 } 00065 00066 virtual cv::MatND getDescriptor() const 00067 { 00068 return descriptor_; 00069 } 00070 00071 protected: 00072 cv::MatND descriptor_; 00073 }; 00074 } 00075 #endif // color_histogram_h_DEFINED