mask.h

来自「tracciatore di mani con webcam」· C头文件 代码 · 共 51 行

#ifndef __HANDVU_MASK_H_INCLUDED__
#define __HANDVU_MASK_H_INCLUDED__

#include <vector>
#include <string>
#include <map>
using namespace std;

typedef vector<double> CDoubleVector;


class Mask {
public:
  Mask();
  ~Mask();

  void ParseFrom(string filename);
  const string& GetName() const { return m_name; }
  double GetProb(int x, int y) const;
  int GetWidth() const {return m_width;}
  int GetHeight() const {return m_height;}
  double GetImageAreaRatio() const { return m_image_area_ratio; }
  void SetImageAreaRatio(double r) { m_image_area_ratio = r; }

protected:
  int               m_width, m_height;
  double            m_image_area_ratio;
  string            m_name;
  CDoubleVector     m_probs;
};

typedef map<string, Mask> MaskMap;
typedef MaskMap::const_iterator ConstMaskIt;

class MaskScaler {
public:
  MaskScaler();
  ~MaskScaler();

public:
  ConstMaskIt    m_mask;
  double         m_scale_x;
  double         m_scale_y;
};



#endif // __HANDVU_MASK_H_INCLUDED__