modules/contrib/src/colortracker.cpp

   1 //*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                                License Agreement
  11 //                       For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2008-2011, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of Intel Corporation may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42 #include "precomp.hpp"
  43 #include "opencv2/contrib/hybridtracker.hpp"
  44
  45 using namespace cv;
  46
  47 CvMeanShiftTracker::CvMeanShiftTracker(CvMeanShiftTrackerParams _params) : params(_params)
  48 {
  49 }
  50
  51 CvMeanShiftTracker::~CvMeanShiftTracker()
  52 {
  53 }
  54
  55 void CvMeanShiftTracker::newTrackingWindow(Mat image, Rect selection)
  56 {
  57     hist.release();
  58     int channels[] = { 0, 0 , 1, 1};
  59     float hrange[] = { 0, 180 };
  60     float srange[] = { 0, 1 };
  61     const float* ranges[] = {hrange, srange};
  62
  63     cvtColor(image, hsv, COLOR_BGR2HSV);
  64     inRange(hsv, Scalar(0, 30, MIN(10, 256)), Scalar(180, 256, MAX(10, 256)), mask);
  65
  66     hue.create(hsv.size(), CV_8UC2);
  67     mixChannels(&hsv, 1, &hue, 1, channels, 2);
  68
  69     Mat roi(hue, selection);
  70     Mat mskroi(mask, selection);
  71     int ch[] = {0, 1};
  72     int chsize[] = {32, 32};
  73     calcHist(&roi, 1, ch, mskroi, hist, 1, chsize, ranges);
  74     normalize(hist, hist, 0, 255, CV_MINMAX);
  75
  76     prev_trackwindow = selection;
  77 }
  78
  79 RotatedRect CvMeanShiftTracker::updateTrackingWindow(Mat image)
  80 {
  81     int channels[] = { 0, 0 , 1, 1};
  82     float hrange[] = { 0, 180 };
  83     float srange[] = { 0, 1 };
  84     const float* ranges[] = {hrange, srange};
  85
  86     cvtColor(image, hsv, COLOR_BGR2HSV);
  87     inRange(hsv, Scalar(0, 30, MIN(10, 256)), Scalar(180, 256, MAX(10, 256)), mask);
  88     hue.create(hsv.size(), CV_8UC2);
  89     mixChannels(&hsv, 1, &hue, 1, channels, 2);
  90     int ch[] = {0, 1};
  91     calcBackProject(&hue, 1, ch, hist, backproj, ranges);
  92     backproj &= mask;
  93
  94     prev_trackbox = CamShift(backproj, prev_trackwindow, TermCriteria(CV_TERMCRIT_EPS | CV_TERMCRIT_ITER, 10, 1));
  95     int cols = backproj.cols, rows = backproj.rows, r = (MIN(cols, rows) + 5) / 6;
  96     prev_trackwindow = Rect(prev_trackwindow.x - r, prev_trackwindow.y - r, prev_trackwindow.x + r,
  97             prev_trackwindow.y + r) & Rect(0, 0, cols, rows);
  98
  99     prev_center.x = (float)(prev_trackwindow.x + prev_trackwindow.width / 2);
 100     prev_center.y = (float)(prev_trackwindow.y + prev_trackwindow.height / 2);
 101
 102 #ifdef DEBUG_HYTRACKER
 103     ellipse(image, prev_trackbox, Scalar(0, 0, 255), 1, CV_AA);
 104 #endif
 105
 106     return prev_trackbox;
 107 }
 108
 109 Mat CvMeanShiftTracker::getHistogramProjection(int type)
 110 {
 111     Mat ms_backproj_f(backproj.size(), type);
 112     backproj.convertTo(ms_backproj_f, type);
 113     return ms_backproj_f;
 114 }
 115
 116 void CvMeanShiftTracker::setTrackingWindow(Rect window)
 117 {
 118     prev_trackwindow = window;
 119 }
 120
 121 Rect CvMeanShiftTracker::getTrackingWindow()
 122 {
 123     return prev_trackwindow;
 124 }
 125
 126 RotatedRect CvMeanShiftTracker::getTrackingEllipse()
 127 {
 128     return prev_trackbox;
 129 }
 130
 131 Point2f CvMeanShiftTracker::getTrackingCenter()
 132 {
 133     return prev_center;
 134 }