dl_packInfer.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_PACKINFER
#define DL_PACKINFER
#include "common/Export.h"
#include "dl_definition.h"
#include "dl_result.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
MMIND_DL_SDK_EXPORT StatusCode createImage(const char* imagePath, MMindImage* image);
 
MMIND_DL_SDK_EXPORT void releaseImage(const MMindImage* image);
 
MMIND_DL_SDK_EXPORT StatusCode showImage(const MMindImage* image, const char* windowName);
 
MMIND_DL_SDK_EXPORT StatusCode saveImage(const MMindImage* image, const char* path);
 
MMIND_DL_SDK_EXPORT StatusCode createPackInferEngine(Engine* engine, const char* packPath,
                                                     const BackendType backend,
                                                     const unsigned int deviceId);
 
MMIND_DL_SDK_EXPORT void releasePackInferEngine(const Engine* engine);
 
MMIND_DL_SDK_EXPORT StatusCode setBatchSizeAndFloatPrecision(
    const Engine* engine, const unsigned int batchSize, const FloatPrecisionType floatPrecision,
    const unsigned int moduleIdx);
 
MMIND_DL_SDK_EXPORT StatusCode setUnsupSegParam(const Engine* engine, const float defectThreshold,
                                                const float nonDefectThreshold,
                                                const unsigned int moduleIdx);
 
MMIND_DL_SDK_EXPORT StatusCode infer(const Engine* engine, const MMindImage* images,
                                     const unsigned int imageNum);
 
MMIND_DL_SDK_EXPORT StatusCode getDefectSegmentataionResult(const Engine* engine,
                                                            const unsigned int moduleIdx,
                                                            DefectAndEdgeResult** results,
                                                            unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseDefectSegmentationResult(DefectAndEdgeResult** results,
                                                         unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getObjectAndInstanceResult(const Engine* engine,
                                                          const unsigned int moduleIdx,
                                                          ObjectAndInsResult** results,
                                                          unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseObjectAndInstanceResult(ObjectAndInsResult** results,
                                                        unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getClassificationResult(const Engine* engine,
                                                       const unsigned int moduleIdx,
                                                       ClassificationResult** results,
                                                       unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseClassificationResult(ClassificationResult** results,
                                                     unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getFastPositioningResult(const Engine* engine,
                                                        const unsigned int moduleIdx,
                                                        FastPositioningResult** results,
                                                        unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseFastPositioningResult(FastPositioningResult** results,
                                                      unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getTextRecognitionResult(const Engine* engine,
                                                        const unsigned int moduleIdx,
                                                        TextRecognitionResult** results,
                                                        unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseTextRecognitionResult(TextRecognitionResult** results,
                                                      unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getTextDetectionResult(const Engine* engine,
                                                      const unsigned int moduleIdx,
                                                      TextDetectionResult** results,
                                                      unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseTextDetectionResult(TextDetectionResult** results,
                                                    unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode getUnsupSegResult(const Engine* engine, const unsigned int moduleIdx,
                                                 UnsupSegResult** results, unsigned int* resultNum);
 
MMIND_DL_SDK_EXPORT void releaseUnsupSegResult(UnsupSegResult** results, unsigned int resultNum);
 
MMIND_DL_SDK_EXPORT StatusCode moduleResultVisualization(const Engine* engine,
                                                         const MMindImage* images,
                                                         const unsigned int moduleIdx,
                                                         const unsigned int imageNum);
 
MMIND_DL_SDK_EXPORT StatusCode resultVisualization(const Engine* engine, const MMindImage* images,
                                                   const unsigned int imageNum);
 
#ifdef __cplusplus
}
#endif
 
#endif