fixed enroll

M Taborsky
1 parent f1ce81bd
Showing 5 changed files with 73 additions and 1 deletions
sdk/core/core.cpp
sdk/openbr_plugin.cpp
sdk/openbr_plugin.h
sdk/plugins/eigen3.cpp
sdk/plugins/meta.cpp
@@ -132,7 +132,15 @@ struct AlgorithmCore
                 if (data.isEmpty()) break;
                 const int numFiles = data.size();
  
-                data >> *transform;
+                if (!Globals->backProject){
+                    data >> *transform;
+                } else {
+                    // I don't really know if this makes sense.
+                    //TemplateList dst;
+                    transform->backProject(data, data);
+                    //data = dst;
+                }
+
                 g->writeBlock(data);
                 const FileList newFiles = data.files();
                 fileList.append(newFiles);
@@ -1144,6 +1144,18 @@ static void _project(const Transform *transform, const Template *src, Template *
     }
 }
  
+static void _backProject(const Transform *transform, const Template *dst, Template *src)
+{
+    try {
+        transform->backProject(*dst, *src);
+    } catch (...) {
+        qWarning("Exception triggered when processing %s with transform %s", qPrintable(src->file.flat()), qPrintable(transform->name()));
+        *src = Template(dst->file);
+        src->file.setBool("FTE");
+    }
+}
+
+
 void Transform::project(const TemplateList &src, TemplateList &dst) const
 {
     dst.reserve(src.size());
@@ -1162,6 +1174,18 @@ void Transform::backProject(const Template &amp;dst, Template &amp;src) const
     src = dst;
 }
  
+void Transform::backProject(const TemplateList &dst, TemplateList &src) const
+{
+    src.reserve(dst.size());
+    for (int i=0; i<dst.size(); i++) src.append(Template());
+
+    QList< QFuture<void> > futures;
+    if (Globals->parallelism) futures.reserve(dst.size());
+    for (int i=0; i<dst.size(); i++)
+        if (Globals->parallelism) futures.append(QtConcurrent::run(_backProject, this, &dst[i], &src[i]));
+        else                                                      _backProject (this, &dst[i], &src[i]);
+    if (Globals->parallelism) Globals->trackFutures(futures);
+}
  
 /* Distance - public methods */
 void Distance::train(const TemplateList &templates)
@@ -913,6 +913,7 @@ public:
     virtual void project(const Template &src, Template &dst) const = 0; /*!< \brief Apply the transform. */
     virtual void project(const TemplateList &src, TemplateList &dst) const; /*!< \brief Apply the transform. */
     virtual void backProject(const Template &dst, Template &src) const; /*!< \brief Invert the transform. */
+    virtual void backProject(const TemplateList &dst, TemplateList &src) const; /*!< \brief Invert the transform. */
  
  
  
@@ -52,6 +52,7 @@ public:
     PCA() : keep(0.95), drop(0), whiten(false) {}
  
 private:
+    /*
     void backProject(const Template &src, Template &dst) const
     {
         const cv::Mat &m = src;
@@ -64,6 +65,7 @@ private:
         // Do projection
         outMap = (eVecs * inMap) + mean;
     }
+    */
  
     double residualReconstructionError(const Template &src) const
     {
@@ -136,6 +136,25 @@ class PipeTransform : public MetaTransform
             Transform::project(src, dst);
         }
     }
+
+    void projectBack(const Template &dst, Template &src) const
+    {
+        src = dst;
+        //reverse order in which transforms are processed
+        int length = transforms.length();
+        //foreach (const Transform *f, transforms) {
+        for (int i = 0; i < length; i++){
+            Transform *f = transforms.at(length - i - 1);
+            try {
+                src >> *f;
+            } catch (...) {
+                qWarning("Exception triggered when processing %s with transform %s", qPrintable(dst.file.flat()), qPrintable(f->name()));
+                src = Template(src.file);
+                src.file.setBool("FTE");
+            }
+        }
+    }
+
 };
  
 BR_REGISTER(Transform, PipeTransform)
@@ -193,6 +212,24 @@ class ChainTransform : public MetaTransform
             dst = Simplified(dst);
         }
     }
+
+    void projectBack(const Template &dst, Template &src) const
+    {
+        src = dst;
+        //reverse order in which transforms are processed
+        int length = transforms.length();
+        //foreach (const Transform *f, transforms) {
+        for (int i = 0; i < length; i++){
+            Transform *f = transforms.at(length - i - 1);
+            try {
+                src >> *f;
+            } catch (...) {
+                qWarning("Exception triggered when processing %s with transform %s", qPrintable(dst.file.flat()), qPrintable(f->name()));
+                src = Template(src.file);
+                src.file.setBool("FTE");
+            }
+        }
+    }
 };
  
 BR_REGISTER(Transform, ChainTransform)