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 as published
  13  *  by the Free Software Foundation, either version 3 of the License,
  14  *  or (at your option) any later version.
  15  *
  16  *  folded-ctf is distributed in the hope that it will be useful, but
  17  *  WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19  *  General Public License for more details.
  20  *
  21  *  You should have received a copy of the GNU General Public License
  22  *  along with folded-ctf.  If not, see <http://www.gnu.org/licenses/>.
  23  *
  24  */
  25
  26 #include "pose_cell.h"
  27 #include "global.h"
  28 #include "tools.h"
  29
  30 bool PoseCell::contains(Pose *pose) {
  31   return
  32     _head_xc.contains(pose->_head_xc) &&
  33     _head_yc.contains(pose->_head_yc) &&
  34     _head_radius.contains(pose->_head_radius) &&
  35     _belly_xc.contains(pose->_belly_xc) &&
  36     _belly_yc.contains(pose->_belly_yc);
  37 }
  38
  39 bool PoseCell::negative_for_train(Pose *pose) {
  40   const scalar_t r_tolerance = 3;
  41   scalar_t r = pose->_head_radius;
  42   return
  43     r <= _head_radius.min / r_tolerance ||
  44     r >= _head_radius.max * r_tolerance ||
  45     pose->_head_xc <= _head_xc.min - r ||
  46     pose->_head_xc >= _head_xc.max + r ||
  47     pose->_head_yc <= _head_yc.min - r ||
  48     pose->_head_yc >= _head_yc.max + r;
  49 }
  50
  51 void PoseCell::get_centroid(Pose *pose) {
  52   pose->_bounding_box_xmin = -1;
  53   pose->_bounding_box_ymin = -1;
  54   pose->_bounding_box_xmax = -1;
  55   pose->_bounding_box_ymax = -1;
  56   pose->_head_radius = sqrt(_head_radius.min * _head_radius.max);
  57   pose->_head_xc = _head_xc.middle();
  58   pose->_head_yc = _head_yc.middle();
  59   pose->_belly_xc = _belly_xc.middle();
  60   pose->_belly_yc = _belly_yc.middle();
  61 }
  62
  63 void PoseCell::print(ostream *out) {
  64   (*out) << "  _head_xc " << _head_xc << endl;
  65   (*out) << "  _head_yc " << _head_yc << endl;
  66   (*out) << "  _head_radius " << _head_radius << endl;
  67   (*out) << "  _head_tilt " << _head_tilt << endl;
  68   (*out) << "  _belly_xc " << _belly_xc << endl;
  69   (*out) << "  _belly_yc " << _belly_yc << endl;
  70 }