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
 };
#endif // INDUSTRIAL
 }
 }
}