model.h

Go to the documentation of this file.

#pragma once
#include "API_EXPORT.h"
#include "common.h"
#include "prediction.h"
#include <filesystem>
#include "config.h"
namespace DaoAI
{
 namespace DeepLearning 
 {
 class Model
 {
 public:
 DAOAI_API Model(const std::string& model_type, const std::filesystem::path& model_file_path, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API virtual ~Model();
 
 DAOAI_API std::string inference(const std::string& image);
 
 private:
 std::string model_name;
 };
 
 namespace Vision
 {
 class ObjectDetection final : public Model
 {
 public:
 DAOAI_API ObjectDetection(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API ObjectDetectionResult inference(const std::string& image);
 };
 class InstanceSegmentation final : public Model
 {
 public:
 DAOAI_API InstanceSegmentation(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API InstanceSegmentationResult inference(const std::string& image);
 };
 class KeypointDetection final : public Model
 {
 public:
 DAOAI_API KeypointDetection(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API KeypointDetectionResult inference(const std::string& image);
 };
 class OCR final : public Model
 {
 public:
 DAOAI_API OCR(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API OCRResult inference(const std::string& image);
 };
 class SemanticSegmentation final : public Model
 {
 public:
 DAOAI_API SemanticSegmentation(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API SemanticSegmentationResult inference(const std::string& image);
 };
 class Classification final : public Model
 {
 public:
 DAOAI_API Classification(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API ClassificationResult inference(const std::string& image);
 };
 class SupervisedDefectSegmentation final : public Model
 {
 public:
 DAOAI_API SupervisedDefectSegmentation(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API SupervisedDefectSegmentationResult inference(const std::string& image);
 };
#ifdef INDUSTRIAL
 class UnsupervisedDefectSegmentation final : public Model
 {
 public:
 DAOAI_API UnsupervisedDefectSegmentation(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API UnsupervisedDefectSegmentationResult inference(const std::string& image);
 };
 
 class PresenceChecking final : public Model
 {
 public:
 DAOAI_API PresenceChecking(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API PresenceCheckingResult inference(const std::string& image);
 };
 
 class Positioning final : public Model
 {
 public:
 DAOAI_API Positioning(const std::filesystem::path& model_file, const DeviceType& device = DeviceType::GPU, const int& device_idx = -1);
 DAOAI_API PositioningResult inference(const std::string& image);
 };
#endif // INDUSTRIAL
 }
 }
}