source: foam/trunk/simple-faceident/src/FaceBank.h @ 59

Revision 59, 2.3 KB checked in by dave, 12 years ago (diff)

added load/save

Line 
1// Copyright (C) 2009 foam
2//
3// This program is free software; you can redistribute it and/or modify
4// it under the terms of the GNU General Public License as published by
5// the Free Software Foundation; either version 2 of the License, or
6// (at your option) any later version.
7//
8// This program is distributed in the hope that it will be useful,
9// but WITHOUT ANY WARRANTY; without even the implied warranty of
10// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11// GNU General Public License for more details.
12//
13// You should have received a copy of the GNU General Public License
14// along with this program; if not, write to the Free Software
15// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
16
17#include <iostream>
18#include <map>
19#include <assert.h>
20#include "cv.h"
21
22#ifndef FACE_BANK
23#define FACE_BANK
24
25/////////////////////////////////////////////////////////////////////////////////
26// A face representation for the facebank
27
28class Face
29{
30public:
31        Face(IplImage *image);
32        ~Face();
33
34        // Blends a newly detected face into this image,
35        // an attempt at making it a little more dynamic
36        // needs more testing.
37        void Learn(const IplImage *image, float blend);
38
39        IplImage *m_Image;
40};
41
42/////////////////////////////////////////////////////////////////////////////////
43// A database of detected faces
44
45class FaceBank
46{
47public:
48        // FaceWidth and FaceHeight are the size for the internal stored image of the face for
49        // comparison, ErrorThresh is the error amount which will trigger a new face to be stored
50        FaceBank(unsigned int FaceWidth, unsigned int FaceHeight, float ErrorThresh);
51       
52        ~FaceBank();
53
54        void Clear();
55       
56        // Learn this face, the face may be a false positive, so we'll discard the
57        // suggestion if we've seen it before, and the error is greater than ErrorThresh
58        float Suggest(IplImage *face, unsigned int ID);
59
60        // Gives the id, given a face, and returns the confidence
61        float Identify(IplImage *face, unsigned int &ID);
62       
63        std::map<unsigned int, Face*> &GetFaceMap() { return m_FaceMap; }
64       
65        unsigned int GetFaceWidth() { return m_FaceWidth; }
66        unsigned int GetFaceHeight() { return m_FaceHeight; }
67
68        void Save(const std::string &filename) const;
69        void Load(const std::string &filename);
70
71private:       
72
73        unsigned int m_FaceWidth;
74        unsigned int m_FaceHeight;
75        float m_ErrorThresh;
76       
77        std::map<unsigned int, Face*> m_FaceMap;
78};
79
80#endif
Note: See TracBrowser for help on using the repository browser.