【Avnet ZU1试用活动】实现车牌识别

h12121

实现车牌识别

还是接着在上一个工程进行操作
对于车牌识别示例，使用了三个模型来实现
车辆检测 ： ssd_traffic
车牌检测 ： PlateDetect
牌照号码识别 ： platenum
源码如下：

1. ~/vitis_ai_cpp_examples/platerecognition/test_jpeg_platerecognition.cpp
   
2. 
   
3. ~/vitis_ai_cpp_examples/platerecognition/test_video_platerecognition.cpp
   
4. 
   
5. 
   
6. 
   
7. #include <glog/logging.h>
   
8. 
   
9. #include <iostream>
   
10. 
    
11. #include <memory>
    
12. 
    
13. #include <opencv2/core.hpp>
    
14. 
    
15. #include <opencv2/highgui.hpp>
    
16. 
    
17. #include <opencv2/imgproc.hpp>
    
18. 
    
19. #include <vitis/ai/demo.hpp>
    
20. 
    
21. #include <vitis/ai/ssd.hpp>
    
22. 
    
23. #include <vitis/ai/nnpp/ssd.hpp>
    
24. 
    
25. #include <vitis/ai/platedetect.hpp>
    
26. 
    
27. #include <vitis/ai/nnpp/platedetect.hpp>
    
28. 
    
29. #include <vitis/ai/platenum.hpp>
    
30. 
    
31. #include <vitis/ai/nnpp/platenum.hpp>
    
32. 
    
33. #include "./platerecognition.hpp"
    
34. 
    
35. 
    
36. 
    
37. using namespace std;
    
38. 
    
39. cv::Mat
    
40. 
    
41. process_result_faces(cv::Mat &image,
    
42. 
    
43.                    const std::vector<vitis::ai::PlateNumResult> &results,
    
44. 
    
45.                    bool is_jpeg) {
    
46. 
    
47.   for (auto &result : results) {
    
48. 
    
49.     //process_result(image, result, is_jpeg);
    
50. 
    
51.   }
    
52. 
    
53.   return image;
    
54. 
    
55. }
    
56. 
    
57. 
    
58. 
    
59. using namespace std;
    
60. 
    
61. int main(int argc, char *argv[]) {
    
62. 
    
63.   return vitis::ai::main_for_jpeg_demo(
    
64. 
    
65.       argc, argv, [] { return vitis::ai::PlateRecognition::create(); },
    
66. 
    
67.       process_result_faces );
    
68. 
    
69. 
    
70. 
    
71. }

复制代码

构建车牌识别应用程序
cd ~/vitis_ai_cpp_examples/platerecognition
./build.sh
在视频上启动车牌识别应用程序
cd ~/vitis_ai_cpp_examples/platerecognition
export DISPLAY=:0.0
xrandr--output DP-1 --mode 800x600
unset PLATERECOGNITION_DEBUG
./test_video_platerecognition ./video/plate_recognition_video.mp4
效果如下：