ArFaceMarker_patch.h

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of OpenSpace3D
For the latest info, see http://www.openspace3d.com
 
Copyright (c) 2012 I-maginer
 
This program is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation; either version 2 of the License, or (at your option) any later
version.
 
This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
 
You should have received a copy of the GNU Lesser General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/copyleft/lesser.txt
 
-----------------------------------------------------------------------------
*/
 
/*
 Toolkit based on OpenCV library
 First version : dec 2010
 Author : Bastien BOURINEAU
*/
 
#ifndef __BTK_FACEMARKER_H__
#define __BTK_FACEMARKER_H__
 
#include "Prerequisites.h"
#include <vector>
#include <iostream>
#include <opencv2/opencv.hpp>
#include "opencv2/features2d/features2d.hpp"
#include "cameraparameters.h"
#include "marker.h"
 
typedef struct model_struct FLANDMARK_Model;
 
class ArFaceMarker : public aruco::Marker
{
public:
  std::vector<cv::Point2f> m_lastCorner;
  bool m_bInitialized;
 
  ArFaceMarker(unsigned int mid);
 
  ~ArFaceMarker();
 
  void calculateFeaturedExtrinsics(float markerSize, aruco::CameraParameters& camParams, bool setYPerperdicular=true) throw(cv::Exception);
 
  void SetImage(const cv::Mat image, std::vector<cv::Rect> objRects);
 
  bool detectMotionFlow(cv::Mat &frame);
 
private:
  unsigned int m_maxFeatures;
  cv::Mat m_image;
  cv::Rect m_ObjRect;
  cv::Mat m_prevFrame;
  std::vector<std::vector<cv::Mat> > m_patches;
  std::vector<cv::Point2f> m_pcorners;
  std::vector<cv::Point2f> m_ObjPoints;
  std::vector<bool> m_lastFound;
  std::vector<cv::Point2f> m_prevFramePoints;
  std::vector<cv::Mat> m_prevPyr;
  std::vector<cv::Mat> m_nextPyr;
  unsigned int m_nbInitFrames;
  unsigned int m_initFrames;
  unsigned int m_minInliners;
  float m_matchScale;
 
  bool GetCorners(std::vector<cv::Point2f> trainPoints, std::vector<cv::Point2f> framePoints);
  cv::Point3f UnProjectPoint(cv::Point2f pt, cv::Point2f offset, float width, float height, float radius);
 
  void GeneratePatches(const cv::Mat image, std::vector<cv::Point2f> points, int patchSize);
  void GeneratePatches(const cv::Mat image, std::vector<cv::Rect> objRects);
 
  cv::Rect scaleRect(cv::Rect& rect);
  cv::Point2f scalePoint2f(cv::Point2f& pt);
  cv::Rect unscaleRect(cv::Rect& rect);
  cv::Point2f unscalePoint2f(cv::Point2f& pt);
 
};
 
#endif