pose_cell.cc

   1 /*
   2  *  folded-ctf is an implementation of the folded hierarchy of
   3  *  classifiers for object detection, developed by Francois Fleuret
   4  *  and Donald Geman.
   5  *
   6  *  Copyright (c) 2008 Idiap Research Institute, http://www.idiap.ch/
   7  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
   8  *
   9  *  This file is part of folded-ctf.
  10  *
  11  *  folded-ctf is free software: you can redistribute it and/or modify
  12  *  it under the terms of the GNU General Public License version 3 as
  13  *  published by the Free Software Foundation.
  14  *
  15  *  folded-ctf is distributed in the hope that it will be useful, but
  16  *  WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  *  General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with folded-ctf.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  */
  24
  25 #include "pose_cell.h"
  26 #include "global.h"
  27 #include "tools.h"
  28
  29 bool PoseCell::contains(Pose *pose) {
  30   return
  31     _head_xc.contains(pose->_head_xc) &&
  32     _head_yc.contains(pose->_head_yc) &&
  33     _head_radius.contains(pose->_head_radius) &&
  34     _belly_xc.contains(pose->_belly_xc) &&
  35     _belly_yc.contains(pose->_belly_yc);
  36 }
  37
  38 bool PoseCell::negative_for_train(Pose *pose) {
  39   const scalar_t r_tolerance = 3;
  40   scalar_t r = pose->_head_radius;
  41   return
  42     r <= _head_radius.min / r_tolerance ||
  43     r >= _head_radius.max * r_tolerance ||
  44     pose->_head_xc <= _head_xc.min - r ||
  45     pose->_head_xc >= _head_xc.max + r ||
  46     pose->_head_yc <= _head_yc.min - r ||
  47     pose->_head_yc >= _head_yc.max + r;
  48 }
  49
  50 void PoseCell::get_centroid(Pose *pose) {
  51   pose->_bounding_box_xmin = -1;
  52   pose->_bounding_box_ymin = -1;
  53   pose->_bounding_box_xmax = -1;
  54   pose->_bounding_box_ymax = -1;
  55   pose->_head_radius = sqrt(_head_radius.min * _head_radius.max);
  56   pose->_head_xc = _head_xc.middle();
  57   pose->_head_yc = _head_yc.middle();
  58   pose->_belly_xc = _belly_xc.middle();
  59   pose->_belly_yc = _belly_yc.middle();
  60 }
  61
  62 void PoseCell::print(ostream *out) {
  63   (*out) << "  _head_xc " << _head_xc << endl;
  64   (*out) << "  _head_yc " << _head_yc << endl;
  65   (*out) << "  _head_radius " << _head_radius << endl;
  66   (*out) << "  _head_tilt " << _head_tilt << endl;
  67   (*out) << "  _belly_xc " << _belly_xc << endl;
  68   (*out) << "  _belly_yc " << _belly_yc << endl;
  69 }