post_processor.h

Go to the documentation of this file.

 
#ifndef SRC_POST_PROCESSOR_H_
#define SRC_POST_PROCESSOR_H_
 
#include <vector>
 
#include "iir.h"
#include "inference_core.h"
#include "intermediate_structures.h"
 
namespace PostProcessing {
 
class PostProcessor {
 private:
  float confidence_threshold;
  std::vector<IIR::IIRFilter> iir_filters;
  IIR::IIRFilter smoothed_trustworthiness;
  bool first_run = true;
 
 public:
  PostProcessor(float confidence_threshold,
                IIR::SmoothingSettings smoothing_settings);
 
  ProcessedResults run(Inference::InferenceResults inference_core_output);
 
  bool set_confidence_threshold(float confidence_threshold);
 
  float get_confidence_threshold();
};
 
}  // namespace PostProcessing
#endif  // SRC_POST_PROCESSOR_H_