Handling negatives otherwhere

Scott Klum
1 parent 6df13fa0
Showing 4 changed files with 5 additions and 5 deletions
openbr/core/bee.cpp
openbr/core/core.cpp
openbr/openbr_plugin.cpp
openbr/plugins/output/eval.cpp
@@ -230,7 +230,7 @@ void makeMask(const QString &amp;targetInput, const QString &amp;queryInput, const QStri
     const FileList targets = TemplateList::fromGallery(targetInput).files();
     const FileList queries = (queryInput == ".") ? targets : TemplateList::fromGallery(queryInput).files();
     const int partitions = targets.first().get<int>("crossValidate");
-    if (partitions == 0) {
+    if (partitions <= 0) {
         writeMatrix(makeMask(targets, queries), mask, targetInput, queryInput);
     } else {
         if (!mask.contains("%1")) qFatal("Mask file name missing partition number place marker (%%1)");
@@ -246,7 +246,7 @@ void makePairwiseMask(const QString &amp;targetInput, const QString &amp;queryInput, con
     const FileList targets = TemplateList::fromGallery(targetInput).files();
     const FileList queries = (queryInput == ".") ? targets : TemplateList::fromGallery(queryInput).files();
     const int partitions = targets.first().get<int>("crossValidate");
-    if (partitions == 0) {
+    if (partitions <= 0) {
         writeMatrix(makePairwiseMask(targets, queries), mask, targetInput, queryInput);
     } else {
         if (!mask.contains("%1")) qFatal("Mask file name missing partition number place marker (%%1)");
@@ -69,7 +69,7 @@ struct AlgorithmCore
         QScopedPointer<Transform> trainingWrapper(br::wrapTransform(transform.data(), "Stream(readMode=DistributeFrames)"));
         TemplateList data(TemplateList::fromGallery(input));
-        if (Globals->crossValidate > 1)
+        if (abs(Globals->crossValidate) > 1)
             for (int i=data.size()-1; i>=0; i--)
                 if (data[i].file.get<bool>("allPartitions",false) || data[i].file.get<bool>("duplicatePartitions",false))
                     data.removeAt(i);
@@ -434,7 +434,7 @@ TemplateList TemplateList::fromGallery(const br::File &amp;gallery)
         if (gallery.getBool("reduce"))
             newTemplates = newTemplates.reduced();
-        if (Globals->crossValidate > 1)
+        if (abs(Globals->crossValidate) > 1)
             newTemplates = newTemplates.partition("Label");
         if (!templates.isEmpty() && gallery.get<bool>("merge", false)) {
@@ -45,7 +45,7 @@ class evalOutput : public MatrixOutput
         if (data.data) {
             const QString csv = QString(file.name).replace(".eval", ".csv");
-            if ((Globals->crossValidate == 0) || (!crossValidate)) {
+            if ((Globals->crossValidate <= 0) || (!crossValidate)) {
                 Evaluate(data, targetFiles, queryFiles, csv);
             } else {
                 QFutureSynchronizer<float> futures;