sources/gst-plugins/gst-nvinfereval/nvds_eval_accumulator.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 "nvds_eval_ap.h"
 
#include <string>
#include <vector>
#include <map>
#include <set>
#include <utility>
#include <cstdint>
 
struct ClassAccum
{
  std::vector<std::pair<float,float>> preds;     /* (conf, best_iou) per prediction */
  uint32_t total_gt  = 0;   /* GT boxes of this class seen this epoch               */
};
 
struct ModelAccum
{
  std::string model_version;
  std::map<std::string, ClassAccum> per_class;   /* class_label -> ClassAccum */
  uint32_t frame_count    = 0;   /* frames processed for this (role, model)      */
  uint32_t frames_with_gt = 0;   /* frames where GT was non-empty (GT available) */
};
 
struct ModelKey
{
  std::string role;         /* "Primary" | "Shadow"      */
  std::string model_name;   /* logical model name        */
 
  bool operator< (const ModelKey &o) const {
    if (role != o.role) return role < o.role;
    return model_name < o.model_name;
  }
  bool operator== (const ModelKey &o) const {
    return role == o.role && model_name == o.model_name;
  }
};
 
class EvalAccumulator
{
public:
  void add_frame (const std::string              &role,
                  const std::string              &model_name,
                  const std::string              &model_version,
                  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);
 
  void reset ();
 
  const std::map<ModelKey, ModelAccum> &models () const { return models_; }
  uint32_t total_frames () const { return total_frames_; }
 
private:
  std::map<ModelKey, ModelAccum> models_;    /* (role, model_name) -> accum  */
  uint32_t                       total_frames_ = 0;
};