-
Notifications
You must be signed in to change notification settings - Fork 0
/
OCR_DORSAL.cpp
107 lines (61 loc) · 2.65 KB
/
OCR_DORSAL.cpp
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*
* File: OCR_DORSAL.cpp
* Author: jesus
*
* Created on 18 de mayo de 2016, 19:49
*/
/*-----------------------------------------------------------------------------*/
template <class T>
OCR_DORSAL<T>::OCR_DORSAL(Ptr<T> _ocr){
OCR = new Ptr<T>( _ocr );
}
/*-----------------------------------------------------------------------------*/
template <class T>
void OCR_DORSAL<T>::NumberRecognition(cv::Mat segmento, std::string &caracter){
OCR->get()->run(segmento, caracter, &boxes, &words, &confidences, OCR_LEVEL_WORD);
caracter.erase(remove(caracter.begin(), caracter.end(), '\n'), caracter.end());
}
/*-----------------------------------------------------------------------------*/
template <class T>
bool OCR_DORSAL<T>::carga_OCR(){
return ( this->OCR->get() || !this->OCR->empty() );
}
/*-----------------------------------------------------------------------------*/
template <class T>
std::string OCR_DORSAL<T>::ReconocerDorsal(cv::Mat &imgBinaria, std::vector<std::vector<cv::Point>> &contornoCaracteres)
{
if ( carga_OCR() ){
size_t numCaracteres = contornoCaracteres.size();
OCR_DORSAL::vecNumber numero;
std::vector<cv::Rect> boundRect(numCaracteres);
std::string digit_numero;
int maxNumCharFault = (double)numCaracteres * RATECHARFAULT;
for (size_t i = 0; i < numCaracteres; i++ ){
boundRect[i] = cv::boundingRect(cv::Mat(contornoCaracteres[i]));
cv::Mat newImagen = cv::Mat::zeros( imgBinaria.size(), CV_8UC1 );
cv::drawContours( newImagen, contornoCaracteres, i, Scalar(255,255,255), -1, 8, std::vector<Vec4i>(), 0, Point() );
cv::Mat segmentacion = newImagen & imgBinaria;
cv::Mat imagenSegment( segmentacion, boundRect[i]);
cv::copyMakeBorder(imagenSegment.clone(), imagenSegment,15,15,15,15,BORDER_CONSTANT,Scalar(0));
NumberRecognition(imagenSegment, digit_numero);
numero.add(digit_numero, boundRect[i].x);
if ( numero.charFault > maxNumCharFault ){
return std::string();
}
}
numero.ordenar();
std::string Num = numero.get();
std::cout<<"DORSAL: "<<Num<<std::endl;
return Num;
}
else{
return std::string();
}
}
/*-----------------------------------------------------------------------------*/
template <class T>
OCR_DORSAL<T>::~OCR_DORSAL(){
if ( OCR )
delete OCR;
OCR = NULL;
}