modelprofile.h

 
#ifndef MODEL_STATS_H
#define MODEL_STATS_H
 
#include <string>
#include <vector>
#include <map>
#include "saver.h"
#include "generic.h"
#include <kll_sketch.hpp>
#include <frequent_items_sketch.hpp>
 
// Assuming declarations for Saver, distributionBox, ClassificationResult, and YoloDetection
 
typedef datasketches::kll_sketch<float> distributionBox;
typedef std::vector<std::pair<float, int>> ClassificationResults;
typedef datasketches::frequent_items_sketch<std::string> frequent_class_sketch;
 
class ModelProfile {
public:
  ModelProfile(std::string model_id, std::string conf_path,
            int save_interval, int top_classes);
  ~ModelProfile();
 
  int log_classification_model_stats(float inference_latency, const ClassificationResults& results);
 
  //int log_yolov5_model_stats(float inference_latency, const YoloDetections& results);
 
  frequent_class_sketch *sketch1;
  int getNumDistributionBoxes() const;
  const distributionBox& getDistributionBox(unsigned int index) const;
 
#ifndef TEST
  private:
#endif
 
  Saver *saver;
  // Member variables (declarations only, definitions in .cpp file)
  std::string model_id_;
  int top_classes_;
  std::string statSavepath;
  std::string dataSavepath;
  std::map<std::string, std::vector<std::string>> modelConfig; 
  std::vector<float> inference_latency_;
  std::vector<int> no_detections_per_image_;
  std::vector<double> objectnessbox_;
  distributionBox *dBox;
  std::map<int, distributionBox*> model_classes_stat_;
};
 
#endif // MODEL_STATS_H