MMIndResult.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.
 ******************************************************************************/
 
#pragma once
 
#include <string>
#include <vector>
#include "MMindImage.h"
 
namespace mmind::dl {
 
struct MMindPoint
{
    int x; 
    int y; 
};
 
struct MMindBbox
{
    float upperLeftX;      
    float upperLeftY;      
    float lowerRightX;     
    float lowerRightY;     
    float confidence;      
    int labelIndex;        
    std::string labelName; 
};
 
struct MMindContour
{
    size_t contourNum = 0;                              
    std::vector<std::vector<MMindPoint>> contourPoints; 
};
 
struct MMindContourWithLabel
{
    size_t contourNum = 0;                              
    std::vector<std::string> labelNames;                
    std::vector<std::vector<MMindPoint>> contourPoints; 
};
 
struct MMindDefectAndEdgeResult
{
    MMindContourWithLabel mask; 
};
 
struct MMindClassificationResult
{
    std::vector<int> labelIndices;       
    std::vector<std::string> labelNames; 
    std::vector<float> confidences;      
};
 
struct MMindObjectAndInsResult
{
    std::vector<MMindBbox> bboxes; 
    MMindContour insMask;          
};
 
struct MMindFastPositioningResult
{
    std::vector<double> angles;                   
    std::vector<MMindPoint> rotationCenterPoints; 
};
 
struct MMindTextRecognitionResult
{
    std::vector<float> confidences; 
    std::vector<std::string> texts; 
};
 
struct MMindTextDetectionResult
{
    std::vector<float> confidences; 
    MMindContour textContour;       
};
 
struct MMindUnsupSegResult
{
    enum class Label {
        OK,     
        NG,     
        UNKNOWN 
    };
    std::vector<Label> labels; 
    MMindContour mask;         
    MMindImage confidenceMat;  
};
 
struct MMindResult
{
    std::vector<MMindClassificationResult> classificationResults;   
    std::vector<MMindDefectAndEdgeResult> defectAndEdgeResults;     
    std::vector<MMindObjectAndInsResult> objecAndInsResults;        
    std::vector<MMindFastPositioningResult> fastPositioningResults; 
    std::vector<MMindTextRecognitionResult> textRecognitionResults; 
    std::vector<MMindTextDetectionResult> textDetectionResults;     
    std::vector<MMindUnsupSegResult> unsupSegResults;               
};
 
} // namespace mmind::dl