LearnedColor.h
上传用户:lijia5631
上传日期:2008-11-10
资源大小:1214k
文件大小:3k
源码类别:

视频捕捉/采集

开发平台:

MultiPlatform

  1. /**
  2.   * HandVu - a library for computer vision-based hand gesture
  3.   * recognition.
  4.   * Copyright (C) 2004 Mathias Kolsch, matz@cs.ucsb.edu
  5.   *
  6.   * This program is free software; you can redistribute it and/or
  7.   * modify it under the terms of the GNU General Public License
  8.   * as published by the Free Software Foundation; either version 2
  9.   * of the License, or (at your option) any later version.
  10.   *
  11.   * This program is distributed in the hope that it will be useful,
  12.   * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.   * GNU General Public License for more details.
  15.   *
  16.   * You should have received a copy of the GNU General Public License
  17.   * along with this program; if not, write to the Free Software
  18.   * Foundation, Inc., 59 Temple Place - Suite 330, 
  19.   * Boston, MA  02111-1307, USA.
  20.   *
  21.   * $Id: LearnedColor.h,v 1.8 2004/11/11 01:58:12 matz Exp $
  22. **/
  23. #if !defined(__LEARNEDCOLOR_H__INCLUDED_)
  24. #define __LEARNEDCOLOR__INCLUDED_
  25. #include <cv.h>
  26. #include "OpticalFlow.h"
  27. #include "Mask.h"
  28. #include "cubicles.h"
  29. #include "Rect.h"
  30. #pragma warning (disable:4786)
  31. class LearnedColor : public ProbDistrProvider {
  32.  public:
  33.   LearnedColor();
  34.   ~LearnedColor();
  35.  public:
  36.   void Initialize(int width, int height);
  37.   void Backproject(IplImage* rgbImage, const CRect& roi);
  38.   void BuildMap(const IplImage* rgbImage, IplImage* map, const CRect& roi);
  39.   virtual double LookupProb(ColorBGR sample) const;
  40.   virtual void CreateMap(const IplImage* rgbImage, IplImage* map, const CRect& roi) const;
  41.   void TestSegmentation(IplImage* rgbImage, 
  42.                         const CRect& roi,
  43.                         double* fpr, double* dr, bool draw);
  44.   void DrawOverlay(IplImage* rgbImage, int overlay_level, 
  45.                    const CRect& roi);
  46.   void LearnFromGroundTruth(IplImage* rgbImage,
  47.                             const CuScanMatch& match, ConstMaskIt mask);
  48.   void GetMostRightUpBlob(IplImage* rgbImage,
  49.     const CRect& roi, CvPoint2D32f& pos);
  50.  protected:
  51.   void SortSamplesIntoBins(vector<ColorBGR> samples, int* cube, int num_bins);
  52.   void ModifyGroundTruth(const CRect& area, double probability);
  53.   void LearnLookupCube(IplImage* rgbImage, const CRect& bbox);
  54.   void SetGroundTruth(const CuScanMatch& match, ConstMaskIt mask,
  55.                       CRect& bbox);
  56. #ifdef DEBUG
  57.   int GetRGBLookupIndex(int x, int y, int z, int size, int arraylen) const;
  58. #else
  59.   int GetRGBLookupIndex(int x, int y, int z, int size, int) const
  60.     { return z*size*size + y*size + x; }
  61. #endif
  62.   
  63.  protected:
  64.   IplImage*              m_truth_map;
  65.   int                    m_truth_pos;
  66.   int                    m_truth_neg;
  67.   vector<float>          m_rgb_lookup;
  68.   int                    m_rgb_lookup_numbins;
  69.   int                    m_rgb_lookup_binsize;
  70. };
  71. #pragma warning (default:4786)
  72. #endif // __SKINCOLOR_H__INCLUDED_