Removed personal debug statements, added filename filtering to empty gallery

Scott Klum
1 parent e7c3e62c
Showing 6 changed files with 7 additions and 14 deletions
openbr/core/bee.cpp
openbr/core/eval.cpp
openbr/plugins/gallery.cpp
openbr/plugins/gui.cpp
openbr/plugins/output.cpp
openbr/plugins/validate.cpp
@@ -102,7 +102,7 @@ void BEE::writeSigset(const QString &amp;sigset, const br::FileList &amp;files, bool ign
                 if ((key == "Index") || (key == "Subject")) continue;
                 metadata.append(key+"=\""+QtUtils::toString(file.value(key))+"\"");
             }
-        lines.append("\t<biometric-signature name=\"" + file.name +"\">");
+        lines.append("\t<biometric-signature name=\"" + file.get<QString>("Subject",file.fileName()) +"\">");
         lines.append("\t\t<presentation file-name=\"" + file.name + "\" " + metadata.join(" ") + "/>");
         lines.append("\t</biometric-signature>");
     }
@@ -273,11 +273,6 @@ cv::Mat BEE::makeMask(const br::FileList &amp;targets, const br::FileList &amp;queries, 
     QList<int> targetPartitions = targets.crossValidationPartitions();
     QList<int> queryPartitions = queries.crossValidationPartitions();
-    qDebug() << "Targets: " << targetLabels << "\n";
-    qDebug() << "Targets: " << targetPartitions << "\n";
-    qDebug() << "Queries: " << queryLabels << "\n";
-    qDebug() << "Queries: " << queryPartitions << "\n";
-
     Mat mask(queries.size(), targets.size(), CV_8UC1);
     for (int i=0; i<queries.size(); i++) {
         const QString &fileA = queries[i];
@@ -303,8 +298,6 @@ cv::Mat BEE::makeMask(const br::FileList &amp;targets, const br::FileList &amp;queries, 
         }
     }
-    qDebug() << mask.row(0);
-
     return mask;
 }
@@ -119,9 +119,6 @@ float Evaluate(const Mat &amp;simmat, const Mat &amp;mask, const QString &amp;csv)
         }
     }
-    qDebug() << genuineCount;
-    qDebug() << impostorCount;
-
     if (numNaNs > 0) qWarning("Encountered %d NaN scores!", numNaNs);
     if (genuineCount == 0) qFatal("No genuine scores!");
     if (impostorCount == 0) qFatal("No impostor scores!");
@@ -168,7 +168,7 @@ class EmptyGallery : public Gallery
             QRegExp re(regexp);
             re.setPatternSyntax(QRegExp::Wildcard);
             for (int i=templates.size()-1; i>=0; i--) {
-                if (!re.exactMatch(templates[i].file.suffix())) {
+                if (!re.exactMatch(templates[i].file.fileName())) {
                     templates.removeAt(i);
                 }
             }
@@ -215,7 +215,7 @@ public:
             foreach (const QString & s, keys) {
                 if (s.compare("name", Qt::CaseInsensitive) == 0) {
-                    newTitle = newTitle + s + ": " + t.file.name + " ";
+                    newTitle = newTitle + s + ": " + t.file.fileName() + " ";
                 } else if (t.file.contains(s)) {
                     QString out = t.file.get<QString>(s);
                     newTitle = newTitle + s + ": " + out + " ";
@@ -442,8 +442,10 @@ class rankOutput : public MatrixOutput
         typedef QPair<int,int> RankPair;
         foreach (const RankPair &pair, Common::Sort(ranks, false))
+            // pair.first == rank retrieved, pair.second == original position
             lines.append(queryFiles[pair.second].name + " " + QString::number(pair.first) + " " + QString::number(scores[pair.second]) + " " + targetFiles[positions[pair.second]].name);
+
         QtUtils::writeFile(file, lines);
     }
 };
@@ -116,7 +116,8 @@ class FilterDistance : public Distance
         foreach (const QString &key, Globals->filters.keys()) {
             bool keep = false;
             const QString metadata = a.file.get<QString>(key, "");
-            if (metadata.isEmpty() || Globals->filters[key].isEmpty()) continue;
+            if (Globals->filters[key].isEmpty()) continue;
+            if (metadata.isEmpty()) return -std::numeric_limits<float>::max();
             foreach (const QString &value, Globals->filters[key]) {
                 if (metadata == value) {
                     keep = true;