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.
122 lines
4.3 KiB
C++
122 lines
4.3 KiB
C++
//
|
|
// Created by Coleflowers on 20/06/2018.
|
|
//
|
|
#include <phpcpp.h>
|
|
#include <iostream>
|
|
|
|
#include "PlateSegmentation.h"
|
|
#include "CNNRecognizer.h"
|
|
#include "Recognizer.h"
|
|
#include "PlateDetection.h"
|
|
#include "FastDeskew.h"
|
|
#include "FineMapping.h"
|
|
|
|
|
|
std::vector<std::string> chars{"京","沪","津","渝","冀","晋","蒙","辽","吉","黑","苏","浙","皖","闽","赣","鲁","豫","鄂","湘","粤","桂","琼","川","贵","云","藏","陕","甘","青","宁","新","0","1","2","3","4","5","6","7","8","9","A","B","C","D","E","F","G","H","J","K","L","M","N","P","Q","R","S","T","U","V","W","X","Y","Z"};
|
|
|
|
/**
|
|
* 车牌图片识别
|
|
* @param imgpath 图片的绝对路径
|
|
* @return 车牌号 图片路径的图片不存在的话 返回空值
|
|
*/
|
|
cv::String scan(const char *imgpath){
|
|
// 读取图片获取车牌的粗略部分
|
|
cv::Mat image = cv::imread(imgpath);
|
|
if(!image.data) {
|
|
printf("No img data!\n");
|
|
return "";
|
|
}
|
|
pr::PlateDetection plateDetection("/usr/platescan/src/lpr/model/cascade.xml");
|
|
std::vector<pr::PlateInfo> plates;
|
|
plateDetection.plateDetectionRough(image,plates);
|
|
// todo 处理发现的每一个车牌信息
|
|
// 目前只处理了一组
|
|
cv::Mat img;
|
|
for(pr::PlateInfo platex:plates) {
|
|
// 暂时不保存了
|
|
//cv::imwrite("res/mmm.png",platex.getPlateImage());
|
|
img = platex.getPlateImage();
|
|
}
|
|
|
|
// 车牌部分
|
|
cv::Mat image_finemapping = pr::FineMapping::FineMappingVertical(img);
|
|
pr::FineMapping finemapper = pr::FineMapping("/usr/platescan/src/lpr/model/HorizonalFinemapping.prototxt","/usr/platescan/src/lpr/model/HorizonalFinemapping.caffemodel");
|
|
|
|
//
|
|
image_finemapping = pr::fastdeskew(image_finemapping, 5);
|
|
|
|
// image_finemapping = finemapper.FineMappingHorizon(image_finemapping,0,-3);
|
|
// Android 代码里这个是
|
|
// 改了之后部分图片数据精度发生变化 ，比如 字母精度有了，但是汉字又错了
|
|
image_finemapping = finemapper.FineMappingHorizon(image_finemapping,2,5);
|
|
// 暂时不保存了
|
|
//cv::imwrite("res/m222222222.png",image_finemapping);
|
|
|
|
// 设置尺寸 识别
|
|
cv::Mat demo = image_finemapping;
|
|
cv::resize(demo,demo,cv::Size(136,36));
|
|
//cv::imwrite("res/m333333.png",demo);
|
|
|
|
cv::Mat respones;
|
|
pr::PlateSegmentation plateSegmentation("/usr/platescan/src/lpr/model/Segmentation.prototxt","/usr/platescan/src/lpr/model/Segmentation.caffemodel");
|
|
pr::PlateInfo plate;
|
|
plate.setPlateImage(demo);
|
|
std::vector<cv::Rect> rects;
|
|
plateSegmentation.segmentPlatePipline(plate,1,rects);
|
|
plateSegmentation.ExtractRegions(plate,rects);
|
|
|
|
pr::GeneralRecognizer *recognizer = new pr::CNNRecognizer("/usr/platescan/src/lpr/model/CharacterRecognization.prototxt","/usr/platescan/src/lpr/model/CharacterRecognization.caffemodel");
|
|
recognizer->SegmentBasedSequenceRecognition(plate);
|
|
//std::cout<<plate.decodePlateNormal(chars)<<std::endl;
|
|
plate.decodePlateNormal(chars);
|
|
|
|
cv::String txt = plate.getPlateName();
|
|
// printf("res:%s\n", txt.c_str());
|
|
|
|
//
|
|
// 写入结果到文件
|
|
// FILE *fp;
|
|
// fp = fopen(resSaveFile, "wb+");
|
|
// fprintf(fp, "%s", txt.c_str());
|
|
// fclose(fp);
|
|
//
|
|
|
|
delete(recognizer);
|
|
return txt;
|
|
}
|
|
|
|
Php::Value myFunction(Php::Parameters ¶ms) {
|
|
|
|
std::string str = params[0];
|
|
const char *str_c = str.c_str();
|
|
|
|
cv::String res = scan(str_c);
|
|
return res.c_str();
|
|
}
|
|
|
|
|
|
/**
|
|
* tell the compiler that the get_module is a pure C function
|
|
*/
|
|
extern "C" {
|
|
|
|
/**
|
|
* Function that is called by PHP right after the PHP process
|
|
* has started, and that returns an address of an internal PHP
|
|
* strucure with all the details and features of your extension
|
|
*
|
|
* @return void* a pointer to an address that is understood by PHP
|
|
*/
|
|
PHPCPP_EXPORT void *get_module() {
|
|
// static(!) Php::Extension object that should stay in memory
|
|
// for the entire duration of the process (that's why it's static)
|
|
static Php::Extension extension("platescan", "1.0");
|
|
|
|
// @todo add your own functions, classes, namespaces to the extension
|
|
extension.add<myFunction>("platescan", {Php::ByRef("string", Php::Type::String)});
|
|
|
|
// return the extension
|
|
return extension;
|
|
}
|
|
}
|