trackerprofile.h

#ifndef TRACKER_PROFILE_H
#define TRACKER_PROFILE_H
 
#include <string>
#include <map>
#include "saver.h"
#include "generic.h"
#include "kll_sketch.hpp"
 
// Forward declaration of classes or structs if needed
class Saver;
 
// Define KLL sketch type
typedef datasketches::kll_sketch<float> distributionBox;
 
class TrackerProfile {
public:
    TrackerProfile(std::string conf_path, int save_interval);
 
    ~TrackerProfile();
    void log_detection_confidence(float confidence);
 
    void log_track_length(int length);
 
    void log_iou(float iou);
 
private:
    Saver* saver;  
    std::map<std::string, std::string> modelConfig;  
    std::string statSavepath;  
 
    // KLL sketches for tracking statistics
    distributionBox confidence_sketch_;
    distributionBox track_length_sketch_;
    distributionBox iou_sketch_;
 
    void registerStatistics();
 
    // Configurations read from the INI file
    std::map<std::string, std::vector<std::string>> trackerConfig;
 
    void cleanup();
};
 
#endif // TRACKER_PROFILE_H