initial commit

wangrong

2025-01-03 2c4b8838f89bebca97615c5f0bac2c58e0349771

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
81
82
83
84
85
86
/* 
* test code for paddle_ocr 
*/ 
#include <opencv2/core.hpp> 
#include <opencv2/imgcodecs.hpp> 
#include <opencv2/imgproc.hpp> 
#include <opencv2/videoio.hpp> 
#include <opencv2/highgui.hpp> 
#include <opencv2/freetype.hpp> 
#include <iostream> 
#include <vector> 
 
#include "../include/paddleocr.h" 
#include "../include/paddlestructure.h" 
 
using namespace PaddleOCR; 
 
int main(int argc, char **argv) { 
 
  auto det_model_dir = "./vp_data/models/text/ppocr/ch_PP-OCRv3_det_infer"; 
  auto rec_model_dir = "./vp_data/models/text/ppocr/ch_PP-OCRv3_rec_infer"; 
  auto cls_model_dir = "./vp_data/models/text/ppocr/ch_ppocr_mobile_v2.0_cls_infer"; 
  auto rec_char_dict_path = "./vp_data/models/text/ppocr/ppocr_keys_v1.txt"; 
  auto ppocr = PPOCR(det_model_dir, cls_model_dir, rec_model_dir, rec_char_dict_path); 
 
  auto video_path = "./vp_data/test_video/ocr.mp4"; 
  auto capture = cv::VideoCapture(video_path); 
  auto ft2 = cv::freetype::createFreeType2(); 
  ft2->loadFontData("./vp_data/font/NotoSansCJKsc-Medium.otf", 0); 
 
  while (true) { 
    cv::Mat frame; 
    if(!capture.read(frame)) { 
      break; 
    } 
    cv::Mat frame1; 
    frame.copyTo(frame1); 
 
    cv::Mat frame2(frame.rows, frame.cols, frame.type(), cv::Scalar(255, 255, 255)); 
    cv::Mat img_vis(frame.rows * 2, frame.cols, frame.type(), cv::Scalar(255, 255, 255)); 
 
 
    std::vector<cv::Mat> frames; 
    //cv::resize(frame, frame, cv::Size(), 0.5, 0.5); 
    frames.push_back(frame); 
 
    auto start = std::chrono::system_clock::now(); 
    auto ocr_results = ppocr.ocr(frames, true, true, false); 
    auto delta = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now() - start); 
    std::cout << "time cost:===" << delta.count() << std::endl; 
 
    assert(ocr_results.size() == 1); 
 
    auto& ocr_result = ocr_results[0]; 
 
 
    for (int n = 0; n < ocr_result.size(); n++) { 
      cv::Point rook_points[4]; 
      for (int m = 0; m < ocr_result[n].box.size(); m++) { 
        rook_points[m] = 
            cv::Point(int(ocr_result[n].box[m][0]), int(ocr_result[n].box[m][1])); 
      } 
 
      const cv::Point *ppt[1] = {rook_points}; 
      int npt[] = {4}; 
      cv::polylines(frame1, ppt, npt, 1, 1, CV_RGB(0, 255, 0), 2, cv::LINE_AA, 0); 
      cv::polylines(frame2, ppt, npt, 1, 1, CV_RGB(0, 255, 0), 1, cv::LINE_AA, 0); 
      //cv::putText(img_vis, ocr_result[n].text, rook_points[0], 1, 0.8, CV_RGB(0, 255, 0)); 
      ft2->putText(frame2, ocr_result[n].text, rook_points[3], 20, cv::Scalar(255, 0, 0), cv::FILLED, cv::LINE_AA, true); 
    } 
 
    auto roi1 = img_vis(cv::Rect(0, 0, frame.cols, frame.rows)); 
    auto roi2 = img_vis(cv::Rect(0, frame.rows, frame.cols, frame.rows)); 
 
    frame1.copyTo(roi1); 
    frame2.copyTo(roi2); 
 
    //cv::imshow("paddle ocr 2", frame2); 
    //cv::imshow("paddle ocr 1", frame1); 
    cv::resize(img_vis, img_vis, cv::Size(), 0.5, 0.5); 
    cv::imshow("paddle ocr 3", img_vis); 
    cv::waitKey(40); 
  } 
   
  return 1; 
}