You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
3.5 KiB
C++
90 lines
3.5 KiB
C++
7 years ago
|
//
|
||
|
|
// Created by Jack Yu on 28/11/2017.
|
||
|
|
//
|
||
|
|
#include "SegmentationFreeRecognizer.h"
|
||
|
|
|
||
|
|
namespace pr {
|
||
|
|
SegmentationFreeRecognizer::SegmentationFreeRecognizer(std::string prototxt, std::string caffemodel) {
|
||
|
|
net = cv::dnn::readNetFromCaffe(prototxt, caffemodel);
|
||
|
|
}
|
||
|
|
inline int judgeCharRange(int id)
|
||
|
|
{return id<31 || id>63;
|
||
|
|
}
|
||
|
|
std::pair<std::string,float> decodeResults(cv::Mat code_table,std::vector<std::string> mapping_table,float thres)
|
||
|
|
{
|
||
|
|
cv::MatSize mtsize = code_table.size;
|
||
|
|
int sequencelength = mtsize[2];
|
||
|
|
int labellength = mtsize[1];
|
||
|
|
cv::transpose(code_table.reshape(1,1).reshape(1,labellength),code_table);
|
||
|
|
std::string name = "";
|
||
|
|
std::vector<int> seq(sequencelength);
|
||
|
|
std::vector<std::pair<int,float>> seq_decode_res;
|
||
|
|
for(int i = 0 ; i < sequencelength; i++) {
|
||
|
|
float *fstart = ((float *) (code_table.data) + i * labellength );
|
||
|
|
int id = std::max_element(fstart,fstart+labellength) - fstart;
|
||
|
|
seq[i] =id;
|
||
|
|
}
|
||
|
|
|
||
|
|
float sum_confidence = 0;
|
||
|
|
int plate_lenghth = 0 ;
|
||
|
|
for(int i = 0 ; i< sequencelength ; i++)
|
||
|
|
{
|
||
|
|
if(seq[i]!=labellength-1 && (i==0 || seq[i]!=seq[i-1]))
|
||
|
|
{
|
||
|
|
float *fstart = ((float *) (code_table.data) + i * labellength );
|
||
|
|
float confidence = *(fstart+seq[i]);
|
||
|
|
std::pair<int,float> pair_(seq[i],confidence);
|
||
|
|
seq_decode_res.push_back(pair_);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
int i = 0;
|
||
|
|
if (seq_decode_res.size()>1 && judgeCharRange(seq_decode_res[0].first) && judgeCharRange(seq_decode_res[1].first))
|
||
|
|
{
|
||
|
|
i=2;
|
||
|
|
int c = seq_decode_res[0].second<seq_decode_res[1].second;
|
||
|
|
name+=mapping_table[seq_decode_res[c].first];
|
||
|
|
sum_confidence+=seq_decode_res[c].second;
|
||
|
|
plate_lenghth++;
|
||
|
|
}
|
||
|
|
|
||
|
|
for(; i < seq_decode_res.size();i++)
|
||
|
|
{
|
||
|
|
name+=mapping_table[seq_decode_res[i].first];
|
||
|
|
sum_confidence +=seq_decode_res[i].second;
|
||
|
|
plate_lenghth++;
|
||
|
|
}
|
||
|
|
std::pair<std::string,float> res;
|
||
|
|
res.second = sum_confidence/plate_lenghth;
|
||
|
|
res.first = name;
|
||
|
|
return res;
|
||
|
|
|
||
|
|
}
|
||
|
|
std::string decodeResults(cv::Mat code_table,std::vector<std::string> mapping_table)
|
||
|
|
{
|
||
|
|
cv::MatSize mtsize = code_table.size;
|
||
|
|
int sequencelength = mtsize[2];
|
||
|
|
int labellength = mtsize[1];
|
||
|
|
cv::transpose(code_table.reshape(1,1).reshape(1,labellength),code_table);
|
||
|
|
std::string name = "";
|
||
|
|
std::vector<int> seq(sequencelength);
|
||
|
|
for(int i = 0 ; i < sequencelength; i++) {
|
||
|
|
float *fstart = ((float *) (code_table.data) + i * labellength );
|
||
|
|
int id = std::max_element(fstart,fstart+labellength) - fstart;
|
||
|
|
seq[i] =id;
|
||
|
|
}
|
||
|
|
for(int i = 0 ; i< sequencelength ; i++)
|
||
|
|
{
|
||
|
|
if(seq[i]!=labellength-1 && (i==0 || seq[i]!=seq[i-1]))
|
||
|
|
name+=mapping_table[seq[i]];
|
||
|
|
}
|
||
|
|
return name;
|
||
|
|
}
|
||
|
|
std::pair<std::string,float> SegmentationFreeRecognizer::SegmentationFreeForSinglePlate(cv::Mat Image,std::vector<std::string> mapping_table) {
|
||
|
|
cv::transpose(Image,Image);
|
||
|
|
cv::Mat inputBlob = cv::dnn::blobFromImage(Image, 1 / 255.0, cv::Size(40,160));
|
||
|
|
net.setInput(inputBlob, "data");
|
||
|
|
cv::Mat char_prob_mat = net.forward();
|
||
|
|
return decodeResults(char_prob_mat,mapping_table,0.00);
|
||
|
|
}
|
||
|
|
}
|