prediction.h

Go to the documentation of this file.

// Copyright (c) 2024, DaoAI Robotics. All rights reserved.
// Version: 0.0.0
 
#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 ""; }
 
 DAOAI_API virtual std::string toAnnotationJSONString();
#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 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 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 ImageEmbedding final : public Prediction
 {
 public:
 DAOAI_API ImageEmbedding();
 DAOAI_API ImageEmbedding(const std::vector<float>& data, const std::vector<size_t>& shape, const int& image_height, const int& image_width);
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
#endif // !BUILD_INFERENCE_CLIENT
 std::vector<float> data;
 std::vector<size_t> shape;
 };
 
 class AutoSegmentationResult final : public Prediction
 {
 public:
 DAOAI_API AutoSegmentationResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#endif // !BUILD_INFERENCE_CLIENT
 
 Mask mask;
 double confidence = -1;
 };
 
 class SupervisedDefectSegmentationResult final : public Prediction
 {
 public:
 DAOAI_API SupervisedDefectSegmentationResult();
#ifndef BUILD_INFERENCE_CLIENT
 DAOAI_API std::string toJSONString() override;
 DAOAI_API std::string toAnnotationJSONString() override;
#else
 DAOAI_API SupervisedDefectSegmentationResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 
 std::unordered_map<std::string, Mask> masks;
 std::vector<float> pixel_scores;
 std::string decision;
 };
 
#ifdef INDUSTRIAL
 class PositioningResult final : public KeypointDetectionResult
 {
 public:
 DAOAI_API PositioningResult();
#ifndef BUILD_INFERENCE_CLIENT
 PositioningResult(const KeypointDetectionResult& keypoint_result);
 DAOAI_API std::string toAnnotationJSONString() override;
 DAOAI_API std::string toJSONString() override;
#else
 DAOAI_API PositioningResult(const std::string& json_result);
#endif // !BUILD_INFERENCE_CLIENT
 
 double angle = 0;
 Point center;
 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
 };
 
#ifndef BUILD_INFERENCE_CLIENT
 class UnsupervisedDefectSegmentationResult
 {
 public:
 struct Region
 {
 Region(const std::string& label, const double& confidence, const bool& defect, const double* bbox) : label(label), confidence(confidence), defect(defect), bbox{ bbox[0], bbox[1], bbox[2], bbox[3] } {}
 std::string label = "";
 double confidence = -1;
 bool defect = false;
 double bbox[4];
 };
 
 UnsupervisedDefectSegmentationResult() {};
 UnsupervisedDefectSegmentationResult(double confidence, bool defect, int image_height, int image_width) :
 confidence(confidence), defect(defect), image_width(image_width), image_height(image_height) {}
 UnsupervisedDefectSegmentationResult(Image&& mask, double confidence, bool defect, int image_height, int image_width) :
 mask(std::move(mask)), confidence(confidence), defect(defect), image_width(image_width), image_height(image_height) {}
 
 UnsupervisedDefectSegmentationResult(Image&& mask, const std::vector<Region>& region_defects, const double& confidence, const bool& defect, const int& image_height, const int& image_width)
 : mask(std::move(mask)), region_defects(region_defects), confidence(confidence), defect(defect), image_width(image_width), image_height(image_height) {}
 
 UnsupervisedDefectSegmentationResult(const std::vector<Region>& region_defects, const int& image_height, const int& image_width) : region_defects(region_defects), image_width(image_width), image_height(image_height) {}
 DAOAI_API std::string toJSONString();
 DAOAI_API std::string toAnnotationJSONString();
 DAOAI_API UnsupervisedDefectSegmentationResult(const std::string& json_result);
 
 Image mask;
 double confidence = -1;
 bool defect = false;
 std::vector<Region> region_defects;
 int image_width = -1;
 int image_height = -1;
 };
#endif // !BUILD_INFERENCE_CLIENT
#endif // INDUSTRIAL
 }
 }
}