detector.h

Go to the documentation of this file.

 
#ifndef _DETECTOR_H_
#define _DETECTOR_H_
 
#pragma once
 
#include <memory>
#include <vector>
 
#include "config.h"
#include "image.h"
#include "export.h"
 
 
namespace NeuroDactyl {
    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 NeuroDactyl::BBox bbox() const = 0; 
        virtual NeuroDactyl::Point cpoint() const = 0;
        virtual float direction() const = 0;
 
        virtual ~Detection() = default;
    };
 
    FPSERVERSDK_API NeuroDactyl::Detection* constructDetection(const NeuroDactyl::Image* image,
                                                                int dpi,
                                                                NeuroDactyl::Point& center,
                                                                float angle);
 
    class Detector {
    public:
        virtual void detect(const NeuroDactyl::Image* image,
                            int dpi,
                            float confidence,
                            std::vector<NeuroDactyl::Detection*>& detections) = 0;
 
        virtual ~Detector() = default;
    };
 
    FPSERVERSDK_API NeuroDactyl::Detector* createDetector(const NeuroDactyl::Config* cfg);
}
 
 
#endif /* _DETECTOR_H_ */