PaddleOCR实现车牌识别系统_采用paddleocr 检测车牌号码

效果展示

如何实现

1.OpenCV环境：个人采用的是OpeCV-440-release版本，网上有很多相关教程
2.PaddleOCR: 这里需要3个文件

models:识别模型下载
paddle_inference_install_dir:预测库下载
PaddleOCR-release-2.0：PaddleOCR 2.0源码下载

models下载
预测库下载：2.0.0版本迅雷下载链接
***PaddleOCR-release-2.0:***
接下来在下图文件夹中建立一个build文件夹

利用CMake编译PaddleOCR源码：添加好路径后，依次点击Configure,Genrate即可

点击ocr_system.sln进入项目中，将ocr_system设为启动项

我简单修改了一下main函数，如果采用我的main函数记得同时将在ocr_rec.h和ocr_rec.cpp中的run函数参数列表末尾添加上vector<vector>&result，这里是因为我想让结果返回在main函数里

#include "glog/logging.h"
#include "omp.h"
#include "opencv2/core.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/imgproc.hpp"
#include<opencv2\opencv.hpp>
#include <chrono>
#include <iomanip>
#include <iostream>
#include <vector>
#include <cstring>
#include <fstream>
#include <numeric>
#include <include/config.h>
#include <include/ocr_det.h>
#include <include/ocr_rec.h>

using namespace std;
using namespace cv;
using namespace PaddleOCR;

int main(int argc, char **argv) {
   OCRConfig config("config.txt");
   std::string img_path("22.jpg");
  cv::Mat srcimg = cv::imread(img_path, cv::IMREAD_COLOR);
  if (srcimg.empty())
  {
      cerr << "图片打开失败" << endl;
      return -1;
  }
  resize(srcimg,srcimg,cv::Size(640,640));
  imshow("检测图",srcimg);
  DBDetector det(config.det_model_dir, config.use_gpu, config.gpu_id,
                 config.gpu_mem, config.cpu_math_library_num_threads,
                 config.use_mkldnn, config.max_side_len, config.det_db_thresh,
                 config.det_db_box_thresh, config.det_db_unclip_ratio,
                 config.visualize, config.use_tensorrt, config.use_fp16);
  Classifier *cls = nullptr;
  if (config.use_angle_cls == true) {
    cls = new Classifier(config.cls_model_dir, config.use_gpu, config.gpu_id,
                         config.gpu_mem, config.cpu_math_library_num_threads,
                         config.use_mkldnn, config.cls_thresh,
                         config.use_tensorrt, config.use_fp16);
  }
  CRNNRecognizer rec(config.rec_model_dir, config.use_gpu, config.gpu_id,
                     config.gpu_mem, config.cpu_math_library_num_threads,
                     config.use_mkldnn, config.char_list_file,
                     config.use_tensorrt, config.use_fp16);
  auto start = std::chrono::system_clock::now();
  std::vector<std::vector<std::vector<int>>> boxes;
  det.Run(srcimg, boxes);
  system("cls");
  vector<vector<string>>result;
  rec.Run(boxes, srcimg, cls,result);  
  //输出结果
  auto it_result = result.begin();
  while (it_result != result.end())
  {
      auto it_str = (*it_result).begin();
      while (it_str != (*it_result).end())
      {
          cout << *it_str++;
      }
      cout << endl;
      ++it_result;
  }

  auto end = std::chrono::system_clock::now();
  auto duration = std::chrono::duration_cast<std::chrono::microseconds>(end - start);
  std::cout << "Cost  "
            << double(duration.count()) *
                   std::chrono::microseconds::period::num /
                   std::chrono::microseconds::period::den
            << "s" << std::endl;
  Mat img = imread("ocr_vis.png");
  imshow("检测结果图",img);
  waitKey(0);
  return 0;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

最后想要让我们的代码运行起来还需要一些条件

config.txt中需要修改三处位置，分别指向我们下载好的识别模型

这里为了方便，我将三个模型名字改了一下，因为我将models和ppocr_keys_v1.txt都直接放入放入当前目录了，所以写的是相对路径。ppocr_keys_v1.txt在PaddleOCR-release-2.0\ppocr\utils下，最后我们添加需要识别的图片和在main函数里修改图片路径，点击运行就可以出现文章开头的识别效果啦
更多的功能可以在这基础上进行扩展，如UI，交互等
参考博客：https://blog.csdn.net/stq054188/article/details/114002913