sources/gst-plugins/gst-nvinfereval/nvds_eval_ap.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include "nvds_eval_gt_loader.h"
 
#include <string>
#include <vector>
#include <utility>
#include <cstdint>
 
float compute_iou (float ax1, float ay1, float ax2, float ay2,
                   float bx1, float by1, float bx2, float by2);
 
void match_preds_to_gt (const std::vector<std::string> &preds_label,
                        const std::vector<float>       &preds_conf,
                        const std::vector<float>       &preds_x1,
                        const std::vector<float>       &preds_y1,
                        const std::vector<float>       &preds_x2,
                        const std::vector<float>       &preds_y2,
                        const std::vector<GTBox>       &gt_boxes,
                        std::vector<float>             &out_best_iou,
                        uint32_t                       &fn_count_out);
 
float compute_ap (const std::vector<std::pair<float,float>> &preds_scored,
                  uint32_t total_gt, float iou_thresh);
 
float compute_ap50   (const std::vector<std::pair<float,float>> &preds_scored,
                      uint32_t total_gt);
 
float compute_ap50_95 (const std::vector<std::pair<float,float>> &preds_scored,
                       uint32_t total_gt);