prediction.h

#pragma once
#include "common.h"
 
namespace DaoAI
{
    namespace DeepLearning
    {
        class PredictionImpl;
 
        struct Instance
        {
            const Box *pred_box;
            const double *score;
            const int *pred_class;
            const std::string* pred_text;
            const std::string* pred_class_name;
            const std::vector<Polygon> *pred_mask;
            const Polygon* pred_flex_box;
            const std::vector<Point> *pred_keypoints;
        };
 
        // Prediction is inference method output. It contains all the useful methods for retrieving results of all tasks but only some are useful depending on the model:
        // - Object detection: pred_boxes(), scores(), pred_classes(), pred_class_names(), pred_masks() (if model supports instance segmentation), and pred_keypoints() (if model supports keypoint prediction)
        // - Anomaly detection: scores(), pred_classes() (0: Normal, 1: Anomalous), pred_class_names() ("Normal" or "Anomalous"), and pred_masks().
        // - Classification: flags().
        class Prediction
        {
        public:
            
            DAOAI_API Prediction();
 
            DAOAI_API ~Prediction();
 
            DAOAI_API Prediction(const Prediction& prediction);
 
            Prediction(const PredictionImpl& prediction_impl);
            
            DAOAI_API std::string toJSONString();
 
            DAOAI_API Prediction& operator=(const Prediction& prediction);
 
            DAOAI_API Instance operator[](size_t index) const;
 
            DAOAI_API int size() const;
 
            DAOAI_API void push_back(const Instance& instance);
 
            DAOAI_API const std::vector<Box>& pred_boxes() const;
 
            DAOAI_API const std::vector<double>& scores() const;
 
            DAOAI_API const std::vector<int>& pred_classes() const; 
 
            DAOAI_API const std::vector<std::vector<Polygon>>& pred_masks() const;
 
            DAOAI_API const std::vector<std::string>& pred_class_names() const;
 
            DAOAI_API const std::vector<std::vector<Point>>& pred_keypoints() const;
 
            DAOAI_API const std::vector<std::pair<std::string, double>>& flags() const;
 
            DAOAI_API const bool& is_multilabel() const;
 
            DAOAI_API const std::vector<std::string>& pred_texts() const;
 
            DAOAI_API const std::vector<Polygon>& pred_flex_boxes() const;
 
            DAOAI_API const int& image_width() const;
 
            DAOAI_API const int& image_height() const;
 
        private:
            std::unique_ptr<PredictionImpl> prediction_impl_;
        };
 
    }
}