detections-utils.h

#pragma once
 
#include <opencv2/core.hpp>
#include <vector>
#include "CelanturDetection.h"
#include "CommonParameters.h"
#include "opencv2/core/mat.hpp"
#include <set>
 
 
namespace celantur {
    void serialise_image_metrics_to_json(cv::Mat image, std::vector<CelanturDetection> detections, std::string image_name, std::string input_folder, std::ostream &os);
 
 
    std::vector<CelanturDetection> filter_detections_by_threshold(const std::vector<CelanturDetection>& detections, const ModelThresholds& thresholds);
 
    std::vector<CelanturDetection> filter_detections_by_class_name(const std::vector<CelanturDetection>& detections, const std::set<std::string>& classes);
 
    std::vector<CelanturDetection> filter_detections_by_class(const std::vector<CelanturDetection>& detections, const std::set<CelanturClassId>& classes);
 
    std::string serialise_detections_proto(const std::vector <CelanturDetection>& detections);
    std::vector<CelanturDetection> deserialise_detections_proto(const std::string& string);
 
    cv::Mat visualise_detections(const cv::Mat image, std::vector<CelanturDetection> detections);
}