CompilationParams.h

#pragma once
#include <ostream>
#include <tuple>
#include <map>
#include <any>
#include <string>
#include <iostream>
#include <algorithm>
 
 
namespace celantur {
struct MinOptMaxDims {
    std::tuple<int, int, int> values;
    static MinOptMaxDims from_tuple(std::tuple<int, int, int>&& in) {
        return MinOptMaxDims {
            .values = in
        };
    }
};
 
enum class OptimisationLevel {
    Low,
    Medium,
    Full
};
 
enum class CompilePrecision {
    FP32,
    FP16,
    INT8
};
 
 
enum class InferenceEnginePluginId {
    None,
    TensorRT,
    ONNX,
    OpenVINO,
    Total
};
 
enum class LogSeverity {
    Error,
    Warning,
    Debug
};
 
struct ModelThresholds {
    float face{0.5f};
    float license_plate{0.5f};
    float person{0.5f};
    float vehicle{0.5f};
    
    float min() const {
        return std::min({face, license_plate, person, vehicle});
    }
};
 
typedef std::map<std::string, std::any > InferenceEnginePluginSettings;
 
typedef std::map<std::string, std::any > InferenceEnginePluginCompileSettings;
std::istream& operator>>(std::istream& in, celantur::OptimisationLevel& opt_level); 
std::ostream& operator<<(std::ostream& out, const celantur::OptimisationLevel& opt_level); 
std::istream& operator>>(std::istream& in, celantur::CompilePrecision& precision);
std::ostream& operator<<(std::ostream& out, const celantur::CompilePrecision& precision);
std::istream& operator>>(std::istream& in, celantur::MinOptMaxDims& dims);
std::ostream& operator<<(std::ostream& out, const celantur::MinOptMaxDims& dims);
std::istream& operator>>(std::istream& in, celantur::InferenceEnginePluginId& plugin_id);
std::ostream& operator<<(std::ostream& out, const celantur::InferenceEnginePluginId& plugin_id);
std::istream& operator>>(std::istream& in, celantur::LogSeverity& log_severity);
std::ostream& operator<<(std::ostream& out, const celantur::LogSeverity& log_severity);
std::istream& operator>>(std::istream& in, celantur::MinOptMaxDims& dims);
std::ostream& operator<<(std::ostream& out, const celantur::MinOptMaxDims& dims);
}