type.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <vector>
#include <string>
 
#include "macro.h"
 
namespace mmind {
 
namespace base {
enum class DataType : int32_t {
    kUnknown,
    kFLOAT,
    kHALF,
    kINT8,
    kINT32,
    kINT64,
};
}
 
namespace dl {
enum class BackendType : int32_t {
    Unknown = 0,
    OnnxRT,
    TensorRT,
    OpenVINO,
    OnnxRTC,
};
 
enum class DLAlgoType : int32_t {
    Unknown = 0,
    Classification,
    ObjectDetection,
    InstanceSegmentation,
    InstanceSegmentationV2,
    DefectSegmentation,
    FastLocating,
    TextRecognition,
    TextDetection,
    UnsupSegmentation,
    LargeSegModel,
    FGBGSegmentation,
    PickAnything,
    RegisterClassification,
    SingleCharacterDetection,
    RegisterUnsupSeg,
    RegisterCharacterDetection
};
 
enum class QuickAIMode : int32_t { Registration = 0, Inference };
 
enum class PrecisionType : int32_t { FP16, FP32 };
 
enum class InferDeviceType {
    Unknown = 0,
    CPU = 1,                  
    GPUDefault = 1 << 1,      
    GPUOptimization = 1 << 2, 
};
 
enum class ModelStatus : unsigned {
    Idle,
    Converting,
    ConvertFailed,
    Converted,
    Loading,
    LoadFailed,
    Loaded,
};
 
enum class InferImageType {
    Depth = 0x0,
    Intensity = 0x1,
};
 
struct ModelEfficiencyParam
{
    int batchSize{1};
    PrecisionType precisionType{PrecisionType::FP32};
 
    bool operator==(const ModelEfficiencyParam& o) const noexcept
    {
        return batchSize == o.batchSize && precisionType == o.precisionType;
    }
 
    bool operator!=(const ModelEfficiencyParam& o) const noexcept { return !(*this == o); }
};
 
} // namespace dl
} // namespace mmind