|
|
#ifndef EASYPR_ACCURACY_HPP
|
|
|
#define EASYPR_ACCURACY_HPP
|
|
|
|
|
|
#include <easypr.h>
|
|
|
#include <ctime>
|
|
|
#include <fstream>
|
|
|
#include <list>
|
|
|
#include <memory>
|
|
|
#include "xml/xmlParser.h"
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
namespace easypr {
|
|
|
|
|
|
namespace demo {
|
|
|
|
|
|
int getGroundTruth(map<string, vector<CPlate>>& xmlMap, const char* path) {
|
|
|
|
|
|
#ifdef OS_WINDOWS
|
|
|
XMLNode::setGlobalOptions(XMLNode::char_encoding_GBK);
|
|
|
#endif
|
|
|
XMLNode xMainNode = XMLNode::openFileHelper(path, "tagset");
|
|
|
|
|
|
int n = xMainNode.nChildNode("image");
|
|
|
|
|
|
// this prints the "coefficient" value for all the "NumericPredictor" tags:
|
|
|
for (int i = 0; i < n; i++) {
|
|
|
XMLNode imageNode = xMainNode.getChildNode("image", i);
|
|
|
string imageName = imageNode.getChildNode("imageName").getText();
|
|
|
|
|
|
vector<CPlate> plateVec;
|
|
|
int m = imageNode.getChildNode("taggedRectangles").nChildNode("taggedRectangle");
|
|
|
for (int j = 0; j < m; j++) {
|
|
|
XMLNode plateNode = imageNode.getChildNode("taggedRectangles").getChildNode("taggedRectangle", j);
|
|
|
|
|
|
int x = atoi(plateNode.getAttribute("x"));
|
|
|
int y = atoi(plateNode.getAttribute("y"));
|
|
|
int width = atoi(plateNode.getAttribute("width"));
|
|
|
int height = atoi(plateNode.getAttribute("height"));
|
|
|
int angle = atoi(plateNode.getAttribute("rotation"));
|
|
|
|
|
|
string plateStr = plateNode.getText();
|
|
|
|
|
|
RotatedRect rr(Point2f(x, y), Size2f(width, height), angle);
|
|
|
|
|
|
CPlate plate;
|
|
|
plate.setPlateStr(plateStr);
|
|
|
plate.setPlatePos(rr);
|
|
|
plateVec.push_back(plate);
|
|
|
}
|
|
|
|
|
|
xmlMap[imageName] = plateVec;
|
|
|
}
|
|
|
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int accuracyTest(const char* test_path) {
|
|
|
std::shared_ptr<easypr::Kv> kv(new easypr::Kv);
|
|
|
kv->load("etc/chinese_mapping");
|
|
|
|
|
|
map<string, vector<CPlate>> xmlMap;
|
|
|
string path(test_path);
|
|
|
path = path + "/GroundTruth.xml";
|
|
|
getGroundTruth(xmlMap, path.c_str());
|
|
|
|
|
|
//cout << xmlMap.size() << endl;
|
|
|
//cout << xmlMap.begin()->first << endl;
|
|
|
//vector<CPlate> plateVec = xmlMap.begin()->second;
|
|
|
//cout << plateVec[0].getPlateStr() << endl;
|
|
|
|
|
|
XMLNode xMainNode = XMLNode::createXMLTopNode("tagset");
|
|
|
#ifdef OS_WINDOWS
|
|
|
XMLNode::setGlobalOptions(XMLNode::char_encoding_GBK);
|
|
|
#endif
|
|
|
auto files = Utils::getFiles(test_path);
|
|
|
|
|
|
std::string path_result = "result/Result.xml";
|
|
|
|
|
|
CPlateRecognize pr;
|
|
|
|
|
|
// 设置Debug模式
|
|
|
|
|
|
pr.setDebug(false);
|
|
|
|
|
|
pr.setLifemode(true);
|
|
|
|
|
|
// 设置要处理的一张图片中最多有多少车牌
|
|
|
|
|
|
pr.setMaxPlates(4);
|
|
|
|
|
|
int size = files.size();
|
|
|
|
|
|
if (0 == size) {
|
|
|
cout << "No File Found in general_test/native_test!" << endl;
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
cout << "Begin to test the easypr accuracy!" << endl;
|
|
|
|
|
|
// 总的测试图片数量
|
|
|
|
|
|
int count_all = 0;
|
|
|
|
|
|
// 错误的图片数<E78987><E695B0>
|
|
|
|
|
|
int count_err = 0;
|
|
|
|
|
|
// 未识别的图片数量
|
|
|
|
|
|
int count_norecogin = 0;
|
|
|
|
|
|
std::list<std::string> not_recognized_files;
|
|
|
|
|
|
// 总的字符差距
|
|
|
|
|
|
float diff_all = 0;
|
|
|
|
|
|
// 平均字符差距
|
|
|
|
|
|
float diff_avg = 0;
|
|
|
|
|
|
// 完全匹配的识别次<E588AB><E6ACA1>
|
|
|
|
|
|
float match_count = 0;
|
|
|
|
|
|
// 完全匹配的识别次数所占识别图片中的比<E79A84><E6AF94>
|
|
|
|
|
|
float match_rate = 0;
|
|
|
|
|
|
// 开始和结束时间
|
|
|
|
|
|
time_t begin, end;
|
|
|
time(&begin);
|
|
|
|
|
|
for (int i = 0; i < size; i++) {
|
|
|
string filepath = files[i].c_str();
|
|
|
|
|
|
// EasyPR开始判断车<E696AD><E8BDA6>
|
|
|
|
|
|
Mat src = imread(filepath);
|
|
|
|
|
|
// 如果是非图像文件，直接过<E68EA5><E8BF87>
|
|
|
|
|
|
if (!src.data) continue;
|
|
|
|
|
|
cout << "------------------" << endl;
|
|
|
|
|
|
// 获取真实的车<E79A84><E8BDA6>
|
|
|
|
|
|
string plateLicense = Utils::getFileName(filepath);
|
|
|
cout << kv->get("original_plate") << ":" << plateLicense << endl;
|
|
|
|
|
|
XMLNode xNode = xMainNode.addChild("image");
|
|
|
xNode.addChild("imageName").addText(plateLicense.c_str());
|
|
|
|
|
|
map<string, vector<CPlate>>::iterator it;
|
|
|
|
|
|
it = xmlMap.find(plateLicense);
|
|
|
if (it != xmlMap.end()) {
|
|
|
cout << it->first << endl;
|
|
|
vector<CPlate> plateVec = it->second;
|
|
|
for (auto plate : plateVec) {
|
|
|
cout << plate.getPlateStr() << " (g)" << endl;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
XMLNode rectangleNodes = xNode.addChild("taggedRectangles");
|
|
|
|
|
|
vector<CPlate> plateVec;
|
|
|
|
|
|
int result = pr.plateRecognize(src, plateVec);
|
|
|
|
|
|
//int result = pr.plateRecognizeAsText(src, plateVec);
|
|
|
|
|
|
if (result == 0) {
|
|
|
int num = plateVec.size();
|
|
|
|
|
|
if (num == 0) {
|
|
|
cout << kv->get("empty_plate") << endl;
|
|
|
if (plateLicense != kv->get("empty_plate")) {
|
|
|
not_recognized_files.push_back(plateLicense);
|
|
|
count_norecogin++;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 多车牌使用diff最小的那个记录
|
|
|
|
|
|
|
|
|
int mindiff = 10000;
|
|
|
for (int j = 0; j < num; j++) {
|
|
|
cout << plateVec[j].getPlateStr() << " (" << j + 1 << ")" << endl;
|
|
|
|
|
|
XMLNode rectangleNode = rectangleNodes.addChild("taggedRectangle");
|
|
|
RotatedRect rr = plateVec[j].getPlatePos();
|
|
|
LocateType locateType = plateVec[j].getPlateLocateType();
|
|
|
|
|
|
rectangleNode.addAttribute("x", to_string((int)rr.center.x).c_str());
|
|
|
rectangleNode.addAttribute("y", to_string((int)rr.center.y).c_str());
|
|
|
rectangleNode.addAttribute("width", to_string((int)rr.size.width).c_str());
|
|
|
rectangleNode.addAttribute("height", to_string((int)rr.size.height).c_str());
|
|
|
|
|
|
rectangleNode.addAttribute("rotation", to_string((int)rr.angle).c_str());
|
|
|
rectangleNode.addAttribute("locateType", to_string(locateType).c_str());
|
|
|
rectangleNode.addText(plateVec[j].getPlateStr().c_str());
|
|
|
|
|
|
string colorplate = plateVec[j].getPlateStr();
|
|
|
|
|
|
// 计算"蓝牌:苏E7KU22"中冒号后面的车牌大小"
|
|
|
|
|
|
|
|
|
vector<string> spilt_plate = Utils::splitString(colorplate, ':');
|
|
|
|
|
|
int size = spilt_plate.size();
|
|
|
if (size == 2 && spilt_plate[1] != "") {
|
|
|
int diff = utils::levenshtein_distance(plateLicense,
|
|
|
spilt_plate[size - 1]);
|
|
|
if (diff < mindiff) mindiff = diff;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
cout << kv->get("diff") << ":" << mindiff << kv->get("char") << endl;
|
|
|
if (mindiff == 0) {
|
|
|
|
|
|
// 完全匹配
|
|
|
|
|
|
match_count++;
|
|
|
}
|
|
|
diff_all = diff_all + mindiff;
|
|
|
}
|
|
|
|
|
|
// 单车牌只计算一次diff
|
|
|
|
|
|
|
|
|
for (int j = 0; j < num; j++) {
|
|
|
cout << plateVec[j].getPlateStr() << endl;
|
|
|
|
|
|
XMLNode rectangleNode = rectangleNodes.addChild("taggedRectangle");
|
|
|
RotatedRect rr = plateVec[j].getPlatePos();
|
|
|
LocateType locateType = plateVec[j].getPlateLocateType();
|
|
|
|
|
|
rectangleNode.addAttribute("x", to_string((int)rr.center.x).c_str());
|
|
|
rectangleNode.addAttribute("y", to_string((int)rr.center.y).c_str());
|
|
|
rectangleNode.addAttribute("width", to_string((int)rr.size.width).c_str());
|
|
|
rectangleNode.addAttribute("height", to_string((int)rr.size.height).c_str());
|
|
|
|
|
|
rectangleNode.addAttribute("rotation", to_string((int)rr.angle).c_str());
|
|
|
rectangleNode.addAttribute("locateType", to_string(locateType).c_str());
|
|
|
rectangleNode.addText(plateVec[j].getPlateStr().c_str());
|
|
|
|
|
|
string colorplate = plateVec[j].getPlateStr();
|
|
|
|
|
|
// 计算"蓝牌:苏E7KU22"中冒号后面的车牌大小"
|
|
|
|
|
|
|
|
|
vector<string> spilt_plate = Utils::splitString(colorplate, ':');
|
|
|
|
|
|
int size = spilt_plate.size();
|
|
|
if (size == 2 && spilt_plate[1] != "") {
|
|
|
int diff = utils::levenshtein_distance(plateLicense,
|
|
|
spilt_plate[size - 1]);
|
|
|
cout << kv->get("diff") << ":" << diff << kv->get("char") << endl;
|
|
|
|
|
|
if (diff == 0) {
|
|
|
|
|
|
// 完全匹配
|
|
|
|
|
|
match_count++;
|
|
|
}
|
|
|
diff_all = diff_all + diff;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
cout << kv->get("error_code") << ":" << result << endl;
|
|
|
cout << kv->get("error_code") << ":" << result << endl;
|
|
|
count_err++;
|
|
|
}
|
|
|
count_all++;
|
|
|
}
|
|
|
time(&end);
|
|
|
|
|
|
cout << "------------------" << endl;
|
|
|
cout << "Easypr accuracy test end!" << endl;
|
|
|
cout << "------------------" << endl;
|
|
|
cout << endl;
|
|
|
cout << kv->get("summaries") << ":" << endl;
|
|
|
cout << kv->get("sum_pictures") << ":" << count_all << ", ";
|
|
|
cout << kv->get("unrecognized") << ":" << count_norecogin << ", ";
|
|
|
|
|
|
xMainNode.writeToFile(path_result.c_str());
|
|
|
|
|
|
float count_recogin = float(count_all - (count_err + count_norecogin));
|
|
|
float count_rate = count_recogin / count_all;
|
|
|
cout << kv->get("locate_rate") << ":" << count_rate * 100 << "% " << endl;
|
|
|
|
|
|
if (count_recogin > 0) {
|
|
|
diff_avg = diff_all / count_recogin;
|
|
|
}
|
|
|
|
|
|
if (count_recogin > 0) {
|
|
|
match_rate = match_count / count_recogin * 100;
|
|
|
}
|
|
|
|
|
|
cout << kv->get("diff_average") << ":" << diff_avg << ", ";
|
|
|
cout << kv->get("full_match") << ":" << match_count << ", ";
|
|
|
cout << kv->get("full_rate") << ":" << match_rate << "% " << endl;
|
|
|
|
|
|
double seconds = difftime(end, begin);
|
|
|
double avgsec = seconds / double(count_all);
|
|
|
|
|
|
cout << kv->get("seconds") << ":" << seconds << kv->get("sec") << ", ";
|
|
|
cout << kv->get("seconds_average") << ":" << avgsec << kv->get("sec") << endl;
|
|
|
|
|
|
cout << kv->get("unrecognized") << ":" << endl;
|
|
|
|
|
|
for (auto it = not_recognized_files.begin(); it != not_recognized_files.end();
|
|
|
++it) {
|
|
|
cout << *it << endl;
|
|
|
}
|
|
|
|
|
|
cout << endl;
|
|
|
|
|
|
cout << "------------------" << endl;
|
|
|
|
|
|
ofstream myfile("accuracy.txt", ios::app);
|
|
|
if (myfile.is_open()) {
|
|
|
time_t t = time(0); // get time now
|
|
|
struct tm* now = localtime(&t);
|
|
|
char buf[80];
|
|
|
|
|
|
strftime(buf, sizeof(buf), "%Y-%m-%d %X", now);
|
|
|
myfile << string(buf) << endl;
|
|
|
|
|
|
myfile << kv->get("sum_pictures") << ":" << count_all << ", ";
|
|
|
myfile << kv->get("unrecognized") << ":" << count_norecogin << ", ";
|
|
|
myfile << kv->get("locate_rate") << ":" << count_rate * 100 << "% "
|
|
|
<< endl;
|
|
|
myfile << kv->get("diff_average") << ":" << diff_avg << ", ";
|
|
|
myfile << kv->get("full_match") << ":" << match_count << ", ";
|
|
|
myfile << kv->get("full_rate") << ":" << match_rate << "% " << endl;
|
|
|
myfile << kv->get("seconds") << ":" << seconds << kv->get("sec") << ", ";
|
|
|
myfile << kv->get("seconds_average") << ":" << avgsec << kv->get("sec")
|
|
|
<< endl;
|
|
|
myfile.close();
|
|
|
}
|
|
|
else {
|
|
|
cout << "Unable to open file";
|
|
|
}
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
}
|
|
|
|
|
|
#endif // EASYPR_ACCURACY_HPP
|
