dl_definition.h

Go to the documentation of this file.

/*******************************************************************************
 *BSD 3-Clause License
 *
 *Copyright (c) 2016-2023, Mech-Mind Robotics
 *All rights reserved.
 *
 *Redistribution and use in source and binary forms, with or without
 *modification, are permitted provided that the following conditions are met:
 *
 *1. Redistributions of source code must retain the above copyright notice, this
 *   list of conditions and the following disclaimer.
 *
 *2. Redistributions in binary form must reproduce the above copyright notice,
 *   this list of conditions and the following disclaimer in the documentation
 *   and/or other materials provided with the distribution.
 *
 *3. Neither the name of the copyright holder nor the names of its
 *   contributors may be used to endorse or promote products derived from
 *   this software without specific prior written permission.
 *
 *THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 *DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 *FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 *DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 *SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 *CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 *OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ******************************************************************************/
 
#ifndef DL_DEFINITION
#define DL_DEFINITION
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef enum StatusCode {
    SUCCESS = 0,               
    E_UNKNOWN,                 
    E_INIT_FAILED,             
    E_INVALID_MODEL_PATH,      
    E_INVALID_MODEL_PACK,      
    E_INVALID_MODEL_VERSION,   
    I_CONVERTING,              
    E_CONVERT_FAILED,          
    E_CONVERT_WITHOUT_INIT,    
    E_UNSUPPORTED_BACKEND,     
    I_LOADING,                 
    E_LOAD_FAILED,             
    E_LOAD_WITHOUT_CONVERT,    
    E_LOAD_WITHOUT_INIT,       
    E_INFER_WITHOUT_LOAD,      
    E_OUT_OF_HOST_MEMORY,      
    E_OUT_OF_DEVICE_MEMORY,    
    E_INFER_FAILED,            
    E_INVALID_IMAGE_PATH,      
    E_UNZIP_ERROR,             
    E_INVALID_IMAGE_NULLPTR,   
    E_MODEL_REGISTEXCEEDLIMIT, 
    E_MODULE_INDEX_ERROR,      
    E_MODULE_TYPE_ERROR,       
    E_SET_GPUID_ERROR,         
    E_INVALID_ENGINE,          
    E_INVALID_BACKEND_TYPE     
} StatusCode;
 
typedef enum BackendType {
    GpuDefault,      
    GpuOptimization, 
    CPU              
} BackendType;
 
typedef enum FloatPrecisionType {
    FP16, 
    FP32  
} FloatPrecisionType;
 
typedef enum DLAlgoType {
    Unknown = 0,
    Classification,  
    ObjectDetection, 
    InstanceSegmentation,
    DefectSegmentation,
    FastLocating,
    EdgeDetection
} DLAlgoType;
 
typedef struct MMindImage
{
    int width;           
    int height;          
    int channel;         
    int depth;           
    unsigned char* data; 
} MMindImage;
 
typedef struct Engine
{
    int type; 
    int id;   
} Engine;
 
#ifdef __cplusplus
}
#endif
 
#endif