detector.h

См. документацию.

 
#ifndef _DETECTOR_H_
#define _DETECTOR_H_
 
#pragma once
 
#include <memory>
#include <vector>
 
#include "config.h"
#include "image.h"
#include "export.h"
 
 
namespace Diametrix {
    struct Point {
        float x;
        float y;
    };
 
    struct BBox {
        float x0;
        float y0;
        float x1;
        float y1;
    };
 
    class Detection {
    public:
        virtual float confidence() const = 0;       
        virtual Diametrix::BBox bbox() const = 0;   
 
        virtual ~Detection() = default;
    };
 
    std::shared_ptr<Diametrix::Detection> DMXSDK_API constructDetection(const std::shared_ptr<Diametrix::Image>& image,
                                                                int dpi,
                                                                Diametrix::Point& center,
                                                                float angle);
 
    class Detector {
    public:
        virtual void detect(const std::shared_ptr<Diametrix::Image>& image,
                            int dpi,
                            float confidence,
                            std::vector<std::shared_ptr<Diametrix::Detection>>& detections) = 0;
 
        virtual ~Detector() = default;
    };
 
    std::unique_ptr<Diametrix::Detector> DMXSDK_API createDetector(const std::shared_ptr<Diametrix::Config>& cfg);
}
 
 
#endif /* _DETECTOR_H_ */