Changeset 94 for foam/trunk/vision

Timestamp:

07/23/2009 01:04:04 PM (11 years ago)

Author:

dave

Message:

added face classifier application, not working yet

Location:

foam/trunk/vision/src

Files:

• Classifier.cpp (modified) (2 diffs)
• Classifier.h (modified) (2 diffs)
• Image.cpp (modified) (1 diff)
• Image.h (modified) (2 diffs)
• PCA.cpp (modified) (1 diff)
• PCA.h (modified) (1 diff)
• PCAClassifier.cpp (modified) (2 diffs)
• PCAClassifier.h (modified) (2 diffs)

foam/trunk/vision/src/Classifier.cpp

@@ -29 +29 @@
 }
 
-void Classifier::AddFeature(int group, const Vector<float> &v) 
+void Classifier::AddFeatureToGroup(int group, const Vector<float> &v) 
 { 
         assert(v.Size()==m_FeatureSize);
         m_Features[group].push_back(v); 
+        // possibly overkill to do this each time we add a new feature...
+        CalcGroupMeans();
 }
 
@@ -50 +52 @@
         m_Mean/=m_FeatureSize;
 }
+
+void Classifier::CalcGroupMeans()
+{
+        for (FeatureMap::iterator i=m_Features.begin();
+                i!=m_Features.end(); ++i)
+        {
+                Vector<float> mean;
+                for (FeatureVec::iterator vi = i->second.begin(); vi!=i->second.end(); ++vi)
+                {
+                        mean+=*vi;
+                }
+                mean/=i->second.size();
+                m_GroupMeans[i->first]=mean;
+        }
+}
+

foam/trunk/vision/src/Classifier.h

@@ -32 +32 @@
         ~Classifier();
         
-        void AddFeature(int group, const Vector<float> &f);
-        virtual int Classify(const Vector<float> &f) = 0;
+        virtual void AddFeature(int group, const Vector<float> &f) = 0;
+        virtual int Classify(const Vector<float> &f, float &error) = 0;
         
 protected:
+        
         void CalcMean();
-        
+        void CalcGroupMeans();
+        void AddFeatureToGroup(int group, const Vector<float> &f);
+
         typedef std::vector<Vector<float> > FeatureVec;
         typedef std::map<int,FeatureVec> FeatureMap;
@@ -47 +50 @@
 
         Vector<float> m_Mean;
-        
+        std::map<int,Vector<float> > m_GroupMeans;
 };
 

foam/trunk/vision/src/Image.cpp

@@ -171 +171 @@
         cvReleaseImage(&m_Image);
         m_Image=newimage;
+}
+
+Image Image::SubImage(int x, int y, int w, int h)
+{
+        CvRect roi;
+        roi.x=x;
+        roi.y=y;
+        roi.width=w;
+        roi.height=h;
+        IplImage *newimage;
+        cvSetImageROI(m_Image,roi);
+        newimage = cvCreateImage( cvSize(roi.width, roi.height), m_Image->depth, m_Image->nChannels );
+        cvCopy(m_Image,newimage);
+        cvReleaseImage(&m_Image);
+        return newimage;
 }
 

foam/trunk/vision/src/Image.h

@@ -25 +25 @@
 {
 public:
+        Image() : m_Image(NULL) {}
         Image(int w, int h, int d, int c);
         Image(const std::string &filename);
@@ -42 +43 @@
         void Crop(int x, int y, int w, int h);
         Image Scale(int w, int h);
+        Image SubImage(int x, int y, int w, int h);
 
         // Paste an image into this one

foam/trunk/vision/src/PCA.cpp

@@ -77 +77 @@
 }
 
-Vector<float> PCA::Project(Vector<float> v)
+Vector<float> PCA::Project(Vector<float> v) const
 {
         return m_EigenTransform*v;
 }
         
-Vector<float> PCA::Synth(Vector<float> v)
+Vector<float> PCA::Synth(Vector<float> v) const
 {
         return m_Mean+m_EigenTransform.VecMulTransposed(v);

foam/trunk/vision/src/PCA.h

@@ -41 +41 @@
         void Compress(unsigned int s, unsigned int e);
         
-        Vector<float> Project(Vector<float> v);
-        Vector<float> Synth(Vector<float> v);
+        Vector<float> Project(Vector<float> v) const;
+        Vector<float> Synth(Vector<float> v) const;
 
         static void RunTests();
 
-        const Vector<float> &GetEigenValues() { return m_EigenValues; }
-        const Matrix<float> &GetEigenTransform() { return m_EigenTransform; }
-        const FeatureVec &GetFeatures() { return m_Features; }
-        const Vector<float> &GetMean() { return m_Mean; }
+        const Vector<float> &GetEigenValues() const { return m_EigenValues; }
+        const Matrix<float> &GetEigenTransform() const { return m_EigenTransform; }
+        const FeatureVec &GetFeatures() const { return m_Features; }
+        const Vector<float> &GetMean() const { return m_Mean; }
+        unsigned int GetFeatureSize() const { return m_FeatureSize; }
         
         

foam/trunk/vision/src/PCAClassifier.cpp

@@ -15 +15 @@
 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 
+#include <cfloat>
 #include "PCAClassifier.h"
 
-PCAClassifier::PCAClassifier(unsigned int FeatureSize) :
-Classifier(FeatureSize)
+using namespace std;
+
+PCAClassifier::PCAClassifier(const PCA &pca) :
+Classifier(pca.GetFeatureSize()),
+m_PCA(pca)
 {
 }
@@ -26 +30 @@
 }
 
+void PCAClassifier::AddFeature(int group, const Vector<float> &f)
+{
+        AddFeatureToGroup(group, m_PCA.Project(f));
+}
 
+int PCAClassifier::Classify(const Vector<float> &f, float &error)
+{
+        Vector<float> params=m_PCA.Project(f);
+
+        // find the closest point in all the group means
+        error = FLT_MAX;
+        int ret=-1;
+        for (map<int,Vector<float> >::iterator i=m_GroupMeans.begin(); 
+                i!=m_GroupMeans.end(); ++i)
+        {
+                float d=params.DistanceFrom(i->second);
+                if (d<error)
+                {
+                        ret=i->first;
+                        error=d;
+                }
+        }
+        
+        return ret;
+}
+
+void PCAClassifier::Refresh()
+{
+        
+}
+

foam/trunk/vision/src/PCAClassifier.h

@@ -21 +21 @@
 #include "Matrix.h"
 #include "Classifier.h"
+#include "PCA.h"
 
 #ifndef FOAM_PCA_CLASSIFIER
@@ -28 +29 @@
 {
 public:
-        PCAClassifier(unsigned int FeatureSize);
+        PCAClassifier(const PCA &pca);
         ~PCAClassifier();
-                
+        
+        // take image vectors in internally uses pca space parameters
+        virtual void AddFeature(int group, const Vector<float> &f);
+        virtual int Classify(const Vector<float> &f, float &error);
+
 private:
-        
-        
+        void Refresh();
+
+        PCA m_PCA;
         
 };