Merge branch 'master' into janus

Josh Klontz
2 parents 6023480a fea31bb9
Showing 19 changed files with 249 additions and 102 deletions
.gitignore
openbr/core/core.cpp
openbr/openbr.cpp
openbr/openbr.h
openbr/openbr_plugin.cpp
openbr/openbr_plugin.h
openbr/plugins/algorithms.cpp
openbr/plugins/crop.cpp
openbr/plugins/draw.cpp
openbr/plugins/gui.cpp
openbr/plugins/landmarks.cpp
openbr/plugins/meta.cpp
openbr/plugins/openbr_internal.h
openbr/plugins/slidingwindow.cpp
openbr/plugins/stream.cpp
openbr/plugins/template.cpp
scripts/evalFaceRecognition-LFW.sh
scripts/evalGenderClassification-PCSO.sh
share/openbr/models
@@ -4,6 +4,7 @@
 data/*/img
 data/*/vid
 data/PCSO/*
+data/lfpw
 build*
 scripts/results
  
@@ -38,3 +39,4 @@ scripts/results
 ### autogenerated sigsets ###
 data/INRIAPerson/sigset
 data/KTH/sigset
+
@@ -70,7 +70,7 @@ struct AlgorithmCore
  
         if (!distance.isNull()) {
             qDebug("Projecting Enrollment");
-            data >> *transform;
+            data >> *downcast;
  
             qDebug("Training Comparison");
             distance->train(data);
@@ -490,9 +490,18 @@ void br::Cat(const QStringList &amp;inputGalleries, const QString &amp;outputGallery)
     }
 }
  
-QSharedPointer<br::Transform> br::Transform::fromAlgorithm(const QString &algorithm)
+QSharedPointer<br::Transform> br::Transform::fromAlgorithm(const QString &algorithm, bool preprocess)
 {
-    return AlgorithmManager::getAlgorithm(algorithm)->transform;
+    if (!preprocess)
+        return AlgorithmManager::getAlgorithm(algorithm)->transform;
+    else {
+        QSharedPointer<Transform> orig_tform = AlgorithmManager::getAlgorithm(algorithm)->transform;
+        QSharedPointer<Transform> newRoot = QSharedPointer<Transform>(Transform::make("Stream(Identity)", NULL));
+        WrapperTransform * downcast = dynamic_cast<WrapperTransform *> (newRoot.data());
+        downcast->transform = orig_tform.data();
+        downcast->init();
+        return newRoot;
+    }
 }
  
 QSharedPointer<br::Distance> br::Distance::fromAlgorithm(const QString &algorithm)
@@ -341,6 +341,12 @@ int br_img_channels(br_template tmpl)
     return t->m().channels();
 }
  
+bool br_img_is_empty(br_template tmpl)
+{
+    Template *t = reinterpret_cast<Template*>(tmpl);
+    return t->m().empty();
+}
+
 void br_set_filename(br_template tmpl, const char *filename)
 {
     Template *t = reinterpret_cast<Template*>(tmpl);
@@ -356,6 +362,12 @@ br_template_list br_enroll_template(br_template tmpl)
     return (br_template_list)tl;
 }
  
+void br_enroll_template_list(br_template_list tl)
+{
+    TemplateList *realTL = reinterpret_cast<TemplateList*>(tl);
+    Enroll(*realTL);
+}
+
 br_template br_get_template(br_template_list tl, int index)
 {
     TemplateList *realTL = reinterpret_cast<TemplateList*>(tl);
@@ -467,6 +467,10 @@ BR_EXPORT int br_img_cols(br_template tmpl);
   */
 BR_EXPORT int br_img_channels(br_template tmpl);
 /*!
+  * \brief Returns if the image is empty.
+  */
+BR_EXPORT bool br_img_is_empty(br_template tmpl);
+/*!
   * \brief Set the filename for a template.
   */
 BR_EXPORT void br_set_filename(br_template tmpl, const char *filename);
@@ -476,6 +480,11 @@ BR_EXPORT void br_set_filename(br_template tmpl, const char *filename);
   */
 BR_EXPORT br_template_list br_enroll_template(br_template tmpl);
 /*!
+  * \brief Enroll a br::TemplateList from the C API!
+  * \param tmpl Pointer to a br::TemplateList.
+  */
+BR_EXPORT void br_enroll_template_list(br_template_list tl);
+/*!
   * \brief Get a pointer to a br::Template at a specified index.
   * \param tl Pointer to a br::TemplateList.
   * \param index The index of the br::Template.
@@ -1184,10 +1184,6 @@ Transform *Transform::make(QString str, QObject *parent)
     if (Globals->abbreviations.contains(str))
         return make(Globals->abbreviations[str], parent);
  
-    { // Check for use of '!' as shorthand for Expand
-        str.replace("!","+Expand+");
-    }
-
     //! [Make a pipe]
     { // Check for use of '+' as shorthand for Pipe(...)
         QStringList words = parse(str, '+');
@@ -1105,7 +1105,7 @@ public:
  
     virtual ~Transform() {}
     static Transform *make(QString str, QObject *parent); /*!< \brief Make a transform from a string. */
-    static QSharedPointer<Transform> fromAlgorithm(const QString &algorithm); /*!< \brief Retrieve an algorithm's transform. */
+    static QSharedPointer<Transform> fromAlgorithm(const QString &algorithm, bool preprocess=true); /*!< \brief Retrieve an algorithm's transform. If preprocess is true, attaches a stream transform as the root of the algorithm*/
  
     virtual Transform *clone() const; /*!< \brief Copy the transform. */
  
@@ -1124,6 +1124,7 @@ public:
  
     /*!< \brief Apply the transform to a single template. Typically used by independent transforms */
     virtual void project(const Template &src, Template &dst) const = 0;
+
     /*!< \brief Apply the transform, taking the full template list as input.
      * A TemplateList is what is typically passed from transform to transform. Transforms that just
      * need to operatoe on a single template at a time (and want to output exactly 1 template) can implement
@@ -1208,7 +1209,9 @@ public:
      * and copy enough of their state that projectUpdate can safely be called on the original
      * instance, and the copy concurrently.
      */
-    virtual Transform * smartCopy() { return this;}
+    virtual Transform * smartCopy(bool & newTransform) { newTransform=false; return this;}
+
+    virtual Transform * smartCopy() {bool junk; return smartCopy(junk);}
  
     /*!
      * \brief Recursively retrieve a named event, returns NULL if an event is not found.
@@ -31,14 +31,14 @@ class AlgorithmsInitializer : public Initializer
     void initialize() const
     {
         // Face
-        Globals->abbreviations.insert("FaceRecognition", "FaceDetection!<FaceRecognitionRegistration>!<FaceRecognitionExtraction>+<FaceRecognitionEmbedding>+<FaceRecognitionQuantization>:MatchProbability(ByteL1)");
-        Globals->abbreviations.insert("GenderClassification", "FaceDetection!<FaceClassificationRegistration>!<FaceClassificationExtraction>+<GenderClassifier>+Discard");
-        Globals->abbreviations.insert("AgeRegression", "FaceDetection!<FaceClassificationRegistration>!<FaceClassificationExtraction>+<AgeRegressor>+Discard");
-        Globals->abbreviations.insert("FaceQuality", "Open!Cascade(FrontalFace)+ASEFEyes+Affine(64,64,0.25,0.35)+ImageQuality+Cvt(Gray)+DFFS+Discard");
-        Globals->abbreviations.insert("MedianFace", "Open!Cascade(FrontalFace)+ASEFEyes+Affine(256,256,0.37,0.45)+Center(Median)");
+        Globals->abbreviations.insert("FaceRecognition", "FaceDetection+Expand+<FaceRecognitionRegistration>+Expand+<FaceRecognitionExtraction>+<FaceRecognitionEmbedding>+<FaceRecognitionQuantization>:MatchProbability(ByteL1)");
+        Globals->abbreviations.insert("GenderClassification", "FaceDetection+Expand+<FaceClassificationRegistration>+Expand+<FaceClassificationExtraction>+<GenderClassifier>+Discard");
+        Globals->abbreviations.insert("AgeRegression", "FaceDetection+Expand+<FaceClassificationRegistration>+Expand+<FaceClassificationExtraction>+<AgeRegressor>+Discard");
+        Globals->abbreviations.insert("FaceQuality", "Open+Expand+Cascade(FrontalFace)+ASEFEyes+Affine(64,64,0.25,0.35)+ImageQuality+Cvt(Gray)+DFFS+Discard");
+        Globals->abbreviations.insert("MedianFace", "Open+Expand+Cascade(FrontalFace)+ASEFEyes+Affine(256,256,0.37,0.45)+Center(Median)");
         Globals->abbreviations.insert("BlurredFaceDetection", "Open+LimitSize(1024)+SkinMask/(Cvt(Gray)+GradientMask)+And+Morph(Erode,16)+LargestConvexArea");
-        Globals->abbreviations.insert("DrawFaceDetection", "Open+Cascade(FrontalFace)!ASEFEyes+Draw");
-        Globals->abbreviations.insert("ShowFaceDetection", "DrawFaceDetection!Show");
+        Globals->abbreviations.insert("DrawFaceDetection", "Open+Cascade(FrontalFace)+Expand+ASEFEyes+Draw");
+        Globals->abbreviations.insert("ShowFaceDetection", "DrawFaceDetection+Expand+Show");
         Globals->abbreviations.insert("OpenBR", "FaceRecognition");
         Globals->abbreviations.insert("GenderEstimation", "GenderClassification");
         Globals->abbreviations.insert("AgeEstimation", "AgeRegression");
@@ -60,7 +60,7 @@ class AlgorithmsInitializer : public Initializer
         Globals->abbreviations.insert("SURF", "Open+KeyPointDetector(SURF)+KeyPointDescriptor(SURF):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("SmallSIFT", "Open+LimitSize(512)+KeyPointDetector(SIFT)+KeyPointDescriptor(SIFT):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("SmallSURF", "Open+LimitSize(512)+KeyPointDetector(SURF)+KeyPointDescriptor(SURF):KeyPointMatcher(BruteForce)");
-        Globals->abbreviations.insert("ColorHist", "Open+LimitSize(512)!EnsureChannels(3)+SplitChannels+Hist(256,0,8)+Cat+Normalize(L1):L2");
+        Globals->abbreviations.insert("ColorHist", "Open+LimitSize(512)+Expand+EnsureChannels(3)+SplitChannels+Hist(256,0,8)+Cat+Normalize(L1):L2");
         Globals->abbreviations.insert("ImageClassification", "Open+CropSquare+LimitSize(256)+Cvt(Gray)+Gradient+Bin(0,360,9,true)+Merge+Integral+RecursiveIntegralSampler(4,2,8,Singleton(KMeans(256)))+Cat+CvtFloat+Hist(256)+KNN(5,Dist(L1),false,5)+Rename(KNN,Subject)");
         Globals->abbreviations.insert("TanTriggs", "Blur(1.1)+Gamma(0.2)+DoG(1,2)+ContrastEq(0.1,10)");
  
@@ -57,10 +57,16 @@ BR_REGISTER(Transform, CropTransform)
 class ROITransform : public UntrainableTransform
 {
     Q_OBJECT
+    Q_PROPERTY(QString propName READ get_propName WRITE set_propName RESET reset_propName STORED false)
+    BR_PROPERTY(QString, propName, "")
  
     void project(const Template &src, Template &dst) const
     {
-        if (src.file.rects().empty()) {
+        if (!propName.isEmpty()) {
+            QRectF rect = src.file.get<QRectF>(propName);
+            dst += src.m()(OpenCVUtils::toRect(rect));
+        }
+        else if (src.file.rects().empty()) {
             dst = src;
             if (Globals->verbose) qWarning("No rects present in file.");
         }
@@ -56,7 +56,7 @@ class DrawTransform : public UntrainableTransform
             for (int i=0; i<pointsList.size(); i++) {
                 const Point2f &point = pointsList[i];
                 circle(dst, point, 3, color, -1);
-                if (verbose) putText(dst, QString::number(i).toStdString(), point, FONT_HERSHEY_SIMPLEX, 0.5, verboseColor, 1);
+                if (verbose) putText(dst, QString("%1,(%2,%3)").arg(QString::number(i),QString::number(point.x),QString::number(point.y)).toStdString(), point, FONT_HERSHEY_SIMPLEX, 0.5, verboseColor, 1);
             }
         }
         if (rects) {
@@ -336,8 +336,7 @@ class AdjacentOverlayTransform : public Transform
  
     void init()
     {
-        opener = br::Transform::fromAlgorithm("Cache(Open)");
-
+        opener = QSharedPointer<br::Transform>(br::Transform::make("Cache(Open)", NULL));
     }
  
 };
@@ -176,6 +176,10 @@ public:
         setFixedSize(200,200);
         QApplication::instance()->installEventFilter(this);
     }
+    ~DisplayWindow()
+    {
+        QApplication::instance()->removeEventFilter(this);
+    }
  
 public slots:
     void showImage(const QPixmap & input)
@@ -193,7 +197,6 @@ public slots:
         setFixedSize(temp);
     }
  
-
     bool eventFilter(QObject * obj, QEvent * event)
     {
         if (event->type() == QEvent::KeyPress)
@@ -420,7 +423,10 @@ public:
     ~ShowTransform()
     {
         delete displayBuffer;
-        delete window;
+        if (QThread::currentThread() == QCoreApplication::instance()->thread())
+            delete window;
+        else
+            emit destroyWindow();
     }
  
     void train(const TemplateList &data) { (void) data; }
@@ -494,6 +500,7 @@ public:
         connect(this, SIGNAL(updateImage(QPixmap)), window,SLOT(showImage(QPixmap)));
         connect(this, SIGNAL(changeTitle(QString)), window, SLOT(setWindowTitle(QString)));
         connect(this, SIGNAL(hideWindow()), window, SLOT(hide()));
+        connect(this, SIGNAL(destroyWindow()), window, SLOT(deleteLater()), Qt::BlockingQueuedConnection);
     }
  
 protected:
@@ -506,6 +513,7 @@ signals:
     void updateImage(const QPixmap & input);
     void changeTitle(const QString & input);
     void hideWindow();
+    void destroyWindow();
 };
 BR_REGISTER(Transform, ShowTransform)
  
@@ -305,6 +305,116 @@ class DrawDelaunayTransform : public UntrainableTransform
  
 BR_REGISTER(Transform, DrawDelaunayTransform)
  
+/*!
+ * \ingroup transforms
+ * \brief Read landmarks from a file and associate them with the correct templates.
+ * \author Scott Klum \cite sklum
+ *
+ * Example of the format:
+ * \code
+ * image_001.jpg:146.000000,190.000000,227.000000,186.000000,202.000000,256.000000
+ * image_002.jpg:75.000000,235.000000,140.000000,225.000000,91.000000,300.000000
+ * image_003.jpg:158.000000,186.000000,246.000000,188.000000,208.000000,233.000000
+ * \endcode
+ */
+class ReadLandmarksTransform : public UntrainableTransform
+{
+    Q_OBJECT
+
+    Q_PROPERTY(QString file READ get_file WRITE set_file RESET reset_file STORED false)
+    Q_PROPERTY(QString imageDelimiter READ get_imageDelimiter WRITE set_imageDelimiter RESET reset_imageDelimiter STORED false)
+    Q_PROPERTY(QString landmarkDelimiter READ get_landmarkDelimiter WRITE set_landmarkDelimiter RESET reset_landmarkDelimiter STORED false)
+    BR_PROPERTY(QString, file, QString())
+    BR_PROPERTY(QString, imageDelimiter, ":")
+    BR_PROPERTY(QString, landmarkDelimiter, ",")
+
+    QHash<QString, QList<QPointF> > landmarks;
+
+    void init()
+    {
+        if (file.isEmpty())
+            return;
+
+        QFile f(file);
+        if (!f.open(QFile::ReadOnly | QFile::Text))
+            qFatal("Failed to open %s for reading.", qPrintable(f.fileName()));
+
+        while (!f.atEnd()) {
+            const QStringList words = QString(f.readLine()).split(imageDelimiter);
+            const QStringList lm = words[1].split(landmarkDelimiter);
+
+            QList<QPointF> points;
+            bool ok;
+            for (int i=0; i<lm.size(); i+=2)
+                points.append(QPointF(lm[i].toFloat(&ok),lm[i+1].toFloat(&ok)));
+            if (!ok) qFatal("Failed to read landmark.");
+
+            landmarks.insert(words[0],points);
+        }
+    }
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+
+        dst.file.appendPoints(landmarks[dst.file.fileName()]);
+    }
+};
+
+BR_REGISTER(Transform, ReadLandmarksTransform)
+
+/*!
+ * \ingroup transforms
+ * \brief Name a point
+ * \author Scott Klum \cite sklum
+ */
+class NamePointsTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(QList<int> indices READ get_indices WRITE set_indices RESET reset_indices STORED false)
+    Q_PROPERTY(QStringList names READ get_names WRITE set_names RESET reset_names STORED false)
+    BR_PROPERTY(QList<int>, indices, QList<int>())
+    BR_PROPERTY(QStringList, names, QStringList())
+
+    void project(const Template &src, Template &dst) const
+    {
+        if (indices.size() != names.size()) qFatal("Point/name size mismatch");
+
+        dst = src;
+
+        QList<QPointF> points = src.file.points();
+
+        for (int i=0; i<indices.size(); i++) {
+            if (indices[i] < points.size()) dst.file.set(names[i], points[indices[i]]);
+            else qFatal("Idex out of range.");
+        }
+    }
+};
+
+BR_REGISTER(Transform, NamePointsTransform)
+
+/*!
+ * \ingroup transforms
+ * \brief Remove a name from a point
+ * \author Scott Klum \cite sklum
+ */
+class AnonymizePointsTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(QStringList names READ get_names WRITE set_names RESET reset_names STORED false)
+    BR_PROPERTY(QStringList, names, QStringList())
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+
+        foreach (const QString &name, names)
+            if (src.file.contains(name)) dst.file.appendPoint(src.file.get<QPointF>(name));
+    }
+};
+
+BR_REGISTER(Transform, AnonymizePointsTransform)
+
 } // namespace br
  
 #include "landmarks.moc"
@@ -248,8 +248,8 @@ class ExpandTransform : public UntrainableMetaTransform
  
     virtual void project(const Template & src, Template & dst) const
     {
-        qFatal("this has gone bad");
-        (void) src; (void) dst;
+        dst = src;
+        qDebug("Called Expand project(Template,Template), nothing will happen");
     }
 };
  
@@ -614,13 +614,20 @@ class DistributeTemplateTransform : public MetaTransform
  
 public:
  
-    Transform * smartCopy()
+    Transform * smartCopy(bool & newTransform)
     {
-        if (!transform->timeVarying())
+        if (!transform->timeVarying()) {
+            newTransform = false;
             return this;
+        }
+        newTransform = true;
  
         DistributeTemplateTransform * output = new DistributeTemplateTransform;
-        output->transform = transform->smartCopy();
+        bool newChild = false;
+        output->transform = transform->smartCopy(newChild);
+        if (newChild)
+            output->transform->setParent(output);
+
         return output;
     }
  
@@ -139,8 +139,9 @@ public:
      *\brief For transforms that don't do any training, this default implementation
      * which creates a new copy of the Transform from its description string is sufficient.
      */
-    virtual Transform * smartCopy()
+    virtual Transform * smartCopy(bool & newTransform)
     {
+        newTransform = true;
         return this->clone();
     }
  
@@ -250,10 +251,13 @@ public:
      * it creates a new copy of its own class, and gives that copy the child transforms
      * returned by calling smartCopy on this transforms children
      */
-    Transform * smartCopy()
+    Transform * smartCopy(bool & newTransform)
     {
-        if (!timeVarying())
+        if (!timeVarying()) {
+            newTransform = false;
             return this;
+        }
+        newTransform = true;
  
         QString name = metaObject()->className();
         name.replace("Transform","");
@@ -266,8 +270,9 @@ public:
  
         foreach(Transform* t, transforms )
         {
-            Transform * maybe_copy = t->smartCopy();
-            if (maybe_copy->parent() == NULL)
+            bool newItem = false;
+            Transform * maybe_copy = t->smartCopy(newItem);
+            if (newItem)
                 maybe_copy->setParent(output);
             output->transforms.append(maybe_copy);
         }
@@ -380,8 +380,10 @@ private:
         QList<Rect> rects;
         QList<float> confidences;
         foreach (const Template &t, src) {
-            rects.append(OpenCVUtils::toRect(t.file.get<QRectF>("Detection")));
-            confidences.append(t.file.get<float>("Confidence"));
+            if (t.file.contains("Detection")) {
+                rects.append(OpenCVUtils::toRect(t.file.get<QRectF>("Detection")));
+                confidences.append(t.file.get<float>("Confidence"));
+            }
         }
  
         // Compute overlap between rectangles and create discrete Laplacian matrix
@@ -1007,14 +1007,24 @@ public:
  
     void project(const Template &src, Template &dst) const
     {
-        (void) src; (void) dst;
-        qFatal("nope");
+        TemplateList in;
+        in.append(src);
+        TemplateList out;
+        CompositeTransform::project(in,out);
+        dst = out.first();
+        if (out.size() > 1)
+            qDebug("Returning first output template only");
     }
  
     void projectUpdate(const Template &src, Template &dst)
     {
-        (void) src; (void) dst;
-        qFatal("whatever");
+        TemplateList in;
+        in.append(src);
+        TemplateList out;
+        projectUpdate(in,out);
+        dst = out.first();
+        if (out.size() > 1)
+            qDebug("Returning first output template only");
     }
  
  
@@ -1082,10 +1092,6 @@ public:
  
         // dst is set to all output received by the final stage, along
         // with anything output via the calls to finalize.
-        //dst = collectionStage->getOutput();
-
-        // dst is set to all output received by the final stage, along
-        // with anything output via the calls to finalize.
         foreach(const TemplateList & list, collector->sets) {
             dst.append(list);
         }
@@ -1368,10 +1374,10 @@ public:
         basis.init();
     }
  
-    Transform * smartCopy()
+    Transform * smartCopy(bool & newTransform)
     {
         // We just want the DirectStream to begin with, so just return a copy of that.
-        DirectStreamTransform * res = (DirectStreamTransform *) basis.smartCopy();
+        DirectStreamTransform * res = (DirectStreamTransform *) basis.smartCopy(newTransform);
         res->activeFrames = this->activeFrames;
         return res;
     }
@@ -50,58 +50,6 @@ class RemoveTemplatesTransform : public UntrainableMetaTransform
  
 BR_REGISTER(Transform, RemoveTemplatesTransform)
  
-/*!
- * \ingroup transforms
- * \brief Name a point
- * \author Scott Klum \cite sklum
- */
-class NamePointsTransform : public UntrainableMetaTransform
-{
-    Q_OBJECT
-    Q_PROPERTY(QList<int> indices READ get_indices WRITE set_indices RESET reset_indices STORED false)
-    Q_PROPERTY(QStringList names READ get_names WRITE set_names RESET reset_names STORED false)
-    BR_PROPERTY(QList<int>, indices, QList<int>())
-    BR_PROPERTY(QStringList, names, QStringList())
-
-    void project(const Template &src, Template &dst) const
-    {
-        if (indices.size() != names.size()) qFatal("Point/name size mismatch");
-
-        dst = src;
-
-        QList<QPointF> points = src.file.points();
-
-        for (int i=0; i<indices.size(); i++) {
-            if (indices[i] < points.size()) dst.file.set(names[i], points[indices[i]]);
-            else qFatal("Idex out of range.");
-        }
-    }
-};
-
-BR_REGISTER(Transform, NamePointsTransform)
-
-/*!
- * \ingroup transforms
- * \brief Remove a name from a point
- * \author Scott Klum \cite sklum
- */
-class AnonymizePointsTransform : public UntrainableMetaTransform
-{
-    Q_OBJECT
-    Q_PROPERTY(QStringList names READ get_names WRITE set_names RESET reset_names STORED false)
-    BR_PROPERTY(QStringList, names, QStringList())
-
-    void project(const Template &src, Template &dst) const
-    {
-        dst = src;
-
-        foreach (const QString &name, names)
-            if (src.file.contains(name)) dst.file.appendPoint(src.file.get<QPointF>(name));
-    }
-};
-
-BR_REGISTER(Transform, AnonymizePointsTransform)
-
 } // namespace br
  
 #include "template.moc"
+#!/bin/bash
+
+ALGORITHM=FaceRecognition
+
+if [ ! -f evalFaceRecognition-LFW.sh ]; then
+  echo "Run this script from the scripts folder!"
+  exit
+fi
+
+if ! hash br 2>/dev/null; then
+  echo "Can't find 'br'. Did you forget to build and install OpenBR? Here's some help: http://openbiometrics.org/doxygen/latest/installation.html"
+  exit
+fi
+
+# Get the data
+./downloadDatasets.sh
+
+if [ ! -e Algorithm_Dataset ]; then
+  mkdir Algorithm_Dataset
+fi
+
+# Run the LFW test protocol
+br -useGui 0 -algorithm $ALGORITHM -path ../data/LFW/img/ -crossValidate 10 -pairwiseCompare ../data/LFW/sigset/test_image_restricted_target.xml ../data/LFW/sigset/test_image_restricted_query.xml ${ALGORITHM}_LFW.mtx -convert Output ${ALGORITHM}_lfw.mtx Algorithm_Dataset/${ALGORITHM}_LFW%1.eval
+
+# Plot results
+br -useGui 0 -plot Algorithm_Dataset/* 'lfw_results.pdf[smooth=Dataset,rocOptions[yLimits=(0,1)]]'
@@ -5,12 +5,11 @@ if [ ! -f evalGenderClassification-PCSO.sh ]; then
 fi
  
 export BR=../build/app/br/br
-export genderAlg=GenderClassification
-
-export PCSO_DIR=/user/pripshare/Databases/FaceDatabases/PCSO/PCSO/
+export ALGORITHM=GenderClassification
+export PCSO_DIR=../data/PCSO/img
  
 # Create a file list by querying the database
 $BR -useGui 0 -quiet -algorithm Identity -enroll "$PCSO_DIR/PCSO.db[query='SELECT File,Gender,PersonID FROM PCSO', subset=1:8000]" terminal.txt > Input.txt
  
 # Enroll the file list and evaluate performance
-$BR -useGui 0  -algorithm $genderAlg -path $PCSO_DIR/Images -enroll Input.txt Output.txt -evalClassification Output.txt Input.txt Gender
 \ No newline at end of file
+$BR -useGui 0  -algorithm $ALGORITHM -path $PCSO_DIR -enroll Input.txt Output.txt -evalClassification Output.txt Input.txt Gender
 \ No newline at end of file
-Subproject commit a73d51013ea05f263e88a28539393159fff2183e
+Subproject commit bcbff8c485f19daddb2e6b2abd5a505ed8c1e526