Commit 9a1a17829fd6769569639f8c1de8c594f10c476b
1 parent
ec7e4126
Use ObjectType instead of Label when getting object types for evaluation. Label …
…is used in other ways throughout the evaluation framework making the output incorrect at times
Showing
1 changed file
with
3 additions
and
3 deletions
openbr/core/evalutils.cpp
| ... | ... | @@ -51,14 +51,14 @@ QList<Detection> EvalUtils::getDetections(const DetectionKey &key, const File &f |
| 51 | 51 | if (pose.contains("Angle")) |
| 52 | 52 | pose = "Frontal"; |
| 53 | 53 | |
| 54 | - QString label = f.get<QString>("Label", ""); | |
| 54 | + QString label = f.get<QString>("ObjectType", "").toLower(); | |
| 55 | 55 | |
| 56 | 56 | const QString filePath = f.path() + "/" + f.fileName(); |
| 57 | 57 | QList<Detection> dets; |
| 58 | 58 | if (key.type == DetectionKey::RectList) { |
| 59 | 59 | QList<QRectF> rects = f.rects(); |
| 60 | 60 | QList<float> confidences = f.getList<float>("Confidences", QList<float>()); |
| 61 | - QList<QString> labels = f.getList<QString>("Labels", QList<QString>()); | |
| 61 | + QList<QString> labels = f.getList<QString>("ObjectTypes", QList<QString>()); | |
| 62 | 62 | if (!isTruth && rects.size() != confidences.size()) |
| 63 | 63 | qFatal("You don't have enough confidence. I mean, your detections don't all have confidence measures."); |
| 64 | 64 | if (!labels.empty() && rects.size() != labels.size()) |
| ... | ... | @@ -263,7 +263,7 @@ QStringList EvalUtils::computeDetectionResults(const QList<ResolvedDetection> &d |
| 263 | 263 | float TP = 0, FP = 0, prevFP = -1, prevTP = -1; |
| 264 | 264 | |
| 265 | 265 | QList<float> FARsToOutput; |
| 266 | - FARsToOutput << 1 << .5 << .2 << .1 << .02 << .01 << .001; | |
| 266 | + FARsToOutput << 10 << 1 << .5 << .2 << .1 << .02 << .01 << .001; | |
| 267 | 267 | |
| 268 | 268 | QDebug debug = qDebug(); |
| 269 | 269 | debug.noquote(); | ... | ... |