Function EvalDetection

openbr/core/eval.cpp:1034–1131

}

float EvalDetection(const QString &predictedGallery, const QString &truthGallery, const QString &csv, bool normalize, int minSize, int maxSize)
{
    qDebug("Evaluating detection of %s against %s", qPrintable(predictedGallery), qPrintable(truthGallery));
    // Organized by file, QMap used to preserve order
    QMap<QString, Detections> allDetections = getDetections(predictedGallery, truthGallery);

    // Remove any bounding boxes with a side smaller than minSize
    if (minSize > 0) {
        qDebug("Removing boxes smaller than %d\n", minSize);
        QMap<QString, Detections> allFilteredDetections;
        foreach (QString key, allDetections.keys()) {
            Detections detections = allDetections[key];
            Detections filteredDetections;
            for (int i = 0; i < detections.predicted.size(); i++) {
                QRectF box = detections.predicted[i].boundingBox;
                if (min(box.width(), box.height()) > sqrt(0.5 * pow(minSize, 2))) {
                    filteredDetections.predicted.append(detections.predicted[i]);
                }
            }
            
            for (int i = 0; i < detections.truth.size(); i++) {
                QRectF box = detections.truth[i].boundingBox;
                if (min(box.width(), box.height()) > minSize) {
                    filteredDetections.truth.append(detections.truth[i]);
                }
            }
            if (!filteredDetections.truth.empty()) allFilteredDetections[key] = filteredDetections;
        }
        allDetections = allFilteredDetections;
    }

    // Remove any bounding boxes with no side smaller than maxSize
    if (maxSize > 0) {
        qDebug("Removing boxes larger than %d\n", maxSize);
        QMap<QString, Detections> allFilteredDetections;
        foreach (QString key, allDetections.keys()) {
            Detections detections = allDetections[key];
            Detections filteredDetections;
            for (int i = 0; i < detections.predicted.size(); i++) {
                QRectF box = detections.predicted[i].boundingBox;
                if (min(box.width(), box.height()) < sqrt(0.5 * pow(maxSize, 2))) {
                    filteredDetections.predicted.append(detections.predicted[i]);
                }
            }

            for (int i = 0; i < detections.truth.size(); i++) {
                QRectF box = detections.truth[i].boundingBox;
                if (min(box.width(), box.height()) < maxSize) {
                    filteredDetections.truth.append(detections.truth[i]);
                }
            }
            if (!filteredDetections.truth.empty()) allFilteredDetections[key] = filteredDetections;
        }
        allDetections = allFilteredDetections;
    }

    QList<ResolvedDetection> resolvedDetections, falseNegativeDetections;
    QRectF normalizations(0, 0, 0, 0);