prediction.h

Go to the documentation of this file.

#pragma once
#include "common.h"
#include <unordered_map>
#include "config.h"
namespace DaoAI
{
 namespace DeepLearning
 {
 class Prediction
 {
 public:
 
#ifndef BUILD_INFERENCE_CLIENT
 virtual std::string toJSONString() { return ""; }
 
 virtual std::string toAnnotationJSONString() { return ""; }
#endif // !BUILD_INFERENCE_CLIENT
 
 
 int image_width;
 int image_height;
 };
 
 
 namespace Vision
 {
 class ObjectDetectionResult : public Prediction
 {
 public:
 DAOAI_API ObjectDetectionResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API ObjectDetectionResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 size_t num_detections = 0;
 std::vector<Box> boxes;
 std::vector<double> confidences;
 std::vector<int> class_ids;
 std::vector<std::string> class_labels;
 std::string decision;
 };
 
 class InstanceSegmentationResult final : public Prediction
 {
 public:
 DAOAI_API InstanceSegmentationResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API InstanceSegmentationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 size_t num_detections = 0;
 std::vector<Box> boxes;
 std::vector<Mask> masks;
 std::vector<double> confidences;
 std::vector<int> class_ids;
 std::vector<std::string> class_labels;
 };
 
 class KeypointDetectionResult : public Prediction
 {
 public:
 DAOAI_API KeypointDetectionResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API KeypointDetectionResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 size_t num_detections = 0;
 std::vector<Box> boxes;
 std::vector<Mask> masks;
 std::vector<std::vector<Point>> keypoints;
 std::vector<double> confidences;
 std::vector<int> class_ids;
 std::vector<std::string> class_labels;
 };
 
 class SemanticSegmentationResult : public Prediction
 {
 public:
 DAOAI_API SemanticSegmentationResult();
 
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
 
#else
 DAOAI_API SemanticSegmentationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 std::unordered_map<std::string, Mask> masks;
 };
 
 class OCRResult final : public Prediction
 {
 public:
 DAOAI_API OCRResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API OCRResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 size_t num_detections = 0;
 std::vector<Polygon> boxes;
 std::vector<double> confidences;
 std::vector<std::string> texts;
 };
 class AnomalyDetectionResult : public Prediction
 {
 public:
 DAOAI_API AnomalyDetectionResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API AnomalyDetectionResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 int label_id = -1;
 std::string label = "";
 Mask mask;
 double confidence = -1;
 };
 class ClassificationResult final : public Prediction
 {
 public:
 DAOAI_API ClassificationResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API ClassificationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 struct Flag
 {
 double confidence = -1;
 std::string label = "";
 };
 
 std::vector<Flag> flags;
 bool multilabel = false;
 };
 
 class SupervisedDefectSegmentationResult final : public SemanticSegmentationResult
 {
 public:
 DAOAI_API SupervisedDefectSegmentationResult();
#ifndef BUILD_INFERENCE_CLIENT
 SupervisedDefectSegmentationResult(const SemanticSegmentationResult& semantic_result);
 DAOAI_API std::string toJSONString() override;
#else
 DAOAI_API SupervisedDefectSegmentationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 std::string decision;
 };
 
 class PositioningResult final : public KeypointDetectionResult
 {
 public:
 DAOAI_API PositioningResult();
#ifndef BUILD_INFERENCE_CLIENT
 PositioningResult(const KeypointDetectionResult& keypoint_result);
 DAOAI_API std::string toJSONString() override;
#else
 DAOAI_API PositioningResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 std::string decision;
 };
 
 class UnsupervisedDefectSegmentationResult final : public AnomalyDetectionResult
 {
 public:
 DAOAI_API UnsupervisedDefectSegmentationResult();
#ifndef BUILD_INFERENCE_CLIENT
 UnsupervisedDefectSegmentationResult(const AnomalyDetectionResult& anomaly_result);
 DAOAI_API std::string toJSONString() override;
#else 
 DAOAI_API UnsupervisedDefectSegmentationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 std::string decision;
 };
 
 class PresenceCheckingResult final : public ObjectDetectionResult
 {
 public:
 DAOAI_API PresenceCheckingResult();
#ifndef BUILD_INFERENCE_CLIENT
 PresenceCheckingResult(const ObjectDetectionResult& object_result);
#else
 DAOAI_API PresenceCheckingResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 };
 }
 }
}