main.cpp 781 Bytes
Newer Older
Your Name's avatar
Your Name committed
1
2
3
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
shizhm's avatar
shizhm committed
4
5
6
7
8
9
#include <SimpleLog.h>
#include <Filesystem.h>
#include <VLPR.h>
#include <fstream>

int main()
Your Name's avatar
Your Name committed
10
{
shizhm's avatar
shizhm committed
11
    // 创建PaddleOCR车牌识别
liucong's avatar
liucong committed
12
13
14
    migraphxSamples::VLPR vlpr;
    migraphxSamples::InitializationParameterOfDB initParamOfDB; 
    migraphxSamples::InitializationParameterOfSVTR initParamOfSVTR;
shizhm's avatar
shizhm committed
15
16
17
    vlpr.Initialize(initParamOfDB, initParamOfSVTR);

    // 读取测试图片
liucong's avatar
liucong committed
18
    cv:: Mat Image=cv::imread("../Resource/Images/vlpr.jpg", 1);
shizhm's avatar
shizhm committed
19
20
21
22
23
24
25
26

    // 推理
    std::vector<std::string> recTexts;
    std::vector<float> recTextScores;
    vlpr.Infer(Image, recTexts, recTextScores); 

    // 打印结果
    for (int i = 0; i < recTexts.size(); i++) 
Your Name's avatar
Your Name committed
27
    {
shizhm's avatar
shizhm committed
28
        printf("VLPR Result:%s\n", recTexts[i].c_str());
Your Name's avatar
Your Name committed
29
30
    }
}