Handle non-thresholdable output for detection plotting

Brendan Klare
1 parent 7f3e4969
Showing 1 changed file with 9 additions and 3 deletions
openbr/core/eval.cpp
@@ -383,13 +383,19 @@ static QStringList computeDetectionResults(const QList&lt;ResolvedDetection&gt; &amp;detec
     }
     const int keep = qMin(points.size(), Max_Points);
-    if (keep < 2) qFatal("Insufficient points.");
+    if (keep < 1) qFatal("Insufficient points.");
     QStringList lines; lines.reserve(keep);
-    for (int i=0; i<keep; i++) {
-        const DetectionOperatingPoint &point = points[double(i) / double(keep-1) * double(points.size()-1)];
+    if (keep == 1) {
+        const DetectionOperatingPoint &point = points[0];
         lines.append(QString("%1ROC, %2, %3").arg(discrete ? "Discrete" : "Continuous", QString::number(point.FalsePositives), QString::number(point.Recall)));
         lines.append(QString("%1PR, %2, %3").arg(discrete ? "Discrete" : "Continuous", QString::number(point.Recall), QString::number(point.Precision)));
+    } else {
+        for (int i=0; i<keep; i++) {
+            const DetectionOperatingPoint &point = points[double(i) / double(keep-1) * double(points.size()-1)];
+            lines.append(QString("%1ROC, %2, %3").arg(discrete ? "Discrete" : "Continuous", QString::number(point.FalsePositives), QString::number(point.Recall)));
+            lines.append(QString("%1PR, %2, %3").arg(discrete ? "Discrete" : "Continuous", QString::number(point.Recall), QString::number(point.Precision)));
+        }
     }
     return lines;
 }