source: foam/trunk/vision/src/PCAClassifier.cpp @ 95

Revision 95, 1.5 KB checked in by dave, 10 years ago (diff)

eigenface user classification seems to work on the test video

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 <cfloat>
18#include "PCAClassifier.h"
19
20using namespace std;
21
22PCAClassifier::PCAClassifier(const PCA &pca) :
23Classifier(pca.GetParamsSize()),
24m_PCA(pca)
25{
26}
27
28PCAClassifier::~PCAClassifier()
29{
30}
31
32void PCAClassifier::AddFeature(int group, const Vector<float> &f)
33{
34        Vector<float> p=m_PCA.Project(f);
35        AddFeatureToGroup(group, p);
36}
37
38int PCAClassifier::Classify(const Vector<float> &f, float &error)
39{
40        Vector<float> params=m_PCA.Project(f);
41
42        // find the closest point in all the group means
43        error = FLT_MAX;
44        int ret=-1;
45        for (map<int,Vector<float> >::iterator i=m_GroupMeans.begin();
46                i!=m_GroupMeans.end(); ++i)
47        {
48                float d=params.DistanceFrom(i->second);
49                if (d<error)
50                {
51                        ret=i->first;
52                        error=d;
53                }
54        }
55       
56        return ret;
57}
58
59void PCAClassifier::Refresh()
60{
61       
62}
63
Note: See TracBrowser for help on using the repository browser.