OcrDB.h

// PaddleOCR车牌检测

#ifndef __OCR_DB_H__
#define __OCR_DB_H__

#include <migraphx/program.hpp>
#include <CommonDefinition.h>
#include <clipper.h>
#include <utility.h>

namespace migraphxSamples
{

typedef struct _DBParameter
{

    float BinaryThreshold;
    float BoxThreshold;
    float UnclipRatio;
    int LimitSideLen;
    std::string ScoreMode;

} DBParameter;

class DB
{
    public:
    DB();

    ~DB();

    ErrorCode Initialize(InitializationParameterOfDB InitializationParameterOfDB);

    ErrorCode Infer(const cv::Mat& img, std::vector<cv::Mat>& imgList);

    private:
    cv::FileStorage configurationFile;

    migraphx::program net;
    cv::Size inputSize;
    std::string inputName;
    migraphx::shape inputShape;
    DBParameter dbParameter;
};

class DBPostProcessor
{
    public:
    void
    GetContourArea(const std::vector<std::vector<float>>& box, float unclip_ratio, float& distance);

    cv::RotatedRect UnClip(std::vector<std::vector<float>> box, const float& unclip_ratio);

    float** Mat2Vec(cv::Mat mat);

    std::vector<std::vector<int>> OrderPointsClockwise(std::vector<std::vector<int>> pts);

    std::vector<std::vector<float>> GetMiniBoxes(cv::RotatedRect box, float& ssid);

    float BoxScoreFast(std::vector<std::vector<float>> box_array, cv::Mat pred);
    float PolygonScoreAcc(std::vector<cv::Point> contour, cv::Mat pred);

    std::vector<std::vector<std::vector<int>>>
    BoxesFromBitmap(const cv::Mat pred,
                    const cv::Mat bitmap,
                    const float& box_thresh,
                    const float& det_db_unclip_ratio,
                    const std::string& det_db_score_mode);

    std::vector<std::vector<std::vector<int>>>
    FilterTagDetRes(std::vector<std::vector<std::vector<int>>> boxes,
                    float ratio_h,
                    float ratio_w,
                    cv::Mat srcimg);

    private:
    static bool XsortInt(std::vector<int> a, std::vector<int> b);

    static bool XsortFp32(std::vector<float> a, std::vector<float> b);

    std::vector<std::vector<float>> Mat2Vector(cv::Mat mat);

    inline int _max(int a, int b) { return a >= b ? a : b; }

    inline int _min(int a, int b) { return a >= b ? b : a; }

    template <class T>
    inline T clamp(T x, T min, T max)
    {
        if(x > max)
            return max;
        if(x < min)
            return min;
        return x;
    }

    inline float clampf(float x, float min, float max)
    {
        if(x > max)
            return max;
        if(x < min)
            return min;
        return x;
    }
};

} // namespace migraphxSamples

#endif