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> deserialise_image_metrics_from_json(std::string_view is);
 
 
    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);
 
 
 
    cv::Mat generate_binary_mask(const cv::Size& image_size, const std::vector<CelanturDetection>& detections, PerTypeProcessingConfig config);
 
    cv::Mat visualise_detections(const cv::Mat image, std::vector<CelanturDetection> detections);
 
    std::string detection_to_yolo_repr(const CelanturDetection& detection, const cv::Size& image_size);
    std::string detections_to_yolo_repr(const std::vector<CelanturDetection>& detections, const cv::Size& image_size);
}