imagesampler.h

 
#ifndef CONFIDENCE_METRICS_H
#define CONFIDENCE_METRICS_H
 
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/opencv.hpp>
#include <vector>
#include <algorithm>
#include <cmath>
#include <fstream>
#include <iostream>
#include <sstream>
#include <stdexcept>
#include <kll_sketch.hpp>
 
#include "iniparser.h"
#include "saver.h"
#include "generic.h"
 
// Typedef for distribution box data structure (assuming datasketches::kll_sketch<unit>)
typedef datasketches::kll_sketch<float> distributionBox;
 
class ImageSampler {
public:
  ImageSampler(const std::string& conf_path, int save_interval, const std::string& model_type);
  ~ImageSampler();
 
   int sample(const void *raw_output, cv::Mat &img, bool save_sample);
  
   float margin_confidence(std::vector<float>& prob_dist, bool sorted);
 
    float least_confidence(std::vector<float>& prob_dist, bool sorted);
 
    float ratio_confidence(std::vector<float>& prob_dist, bool sorted);
 
    float entropy_confidence(std::vector<float>& prob_dist);
    std::string statSavepath;
    std::string dataSavepath;
 
private:   
    // Member variables for storing confidence metric statistics
    distributionBox marginConfidenceBox;
    distributionBox leastConfidenceBox;
    distributionBox ratioConfidenceBox;
    distributionBox entropyConfidenceBox;
    std::string model_type;
    Saver *saver;
    //ImageUploader *uploader;
    std::map<std::string, std::vector<std::string>> samplingConfig;
    void registerStatistics(const std::string& name);
 
    float computeConfidence(const std::string& name, std::vector<float>& confidence);
 
    bool isThresholdExceeded(const std::string& name, float stat_score, const std::vector<std::string>& config);
 
    void updateSamplingStatistics(const std::string& name, float confidence_score);
};
 
#endif // CONFIDENCE_METRICS_H