EasyPR 1.01 update

.gitignore

@@ -13,6 +13,7 @@ Thumbs.db
 [Dd]ebugPublic/
 [Rr]elease/
 [Rr]eleases/
+experiment/
 x64/
 x86/
 build/

EasyPR.vcxproj

@@ -47,11 +47,11 @@
   </PropertyGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
     <ClCompile>
-      <PrecompiledHeader>
-      </PrecompiledHeader>
+      <PrecompiledHeader>NotUsing</PrecompiledHeader>
       <WarningLevel>Level3</WarningLevel>
       <Optimization>Disabled</Optimization>
       <PreprocessorDefinitions>WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions)</PreprocessorDefinitions>
+      <PrecompiledHeaderFile>stdafx.h</PrecompiledHeaderFile>
     </ClCompile>
     <Link>
       <SubSystem>Console</SubSystem>
@@ -78,35 +78,39 @@
   <ItemGroup>
     <None Include="model\ann.xml" />
     <None Include="model\svm.xml" />
-    <None Include="train\训练说明.txt" />
     <None Include="关于版权.txt" />
   </ItemGroup>
   <ItemGroup>
-    <ClCompile Include="src\main\easypr\data_prepare.cpp" />
-    <ClCompile Include="src\main\easypr\data_proc.cpp" />
-    <ClCompile Include="src\main\easypr\easypr_main.cpp" />
-    <ClCompile Include="src\main\core\chars_identify.cpp" />
-    <ClCompile Include="src\main\core\chars_recognise.cpp" />
-    <ClCompile Include="src\main\core\chars_segment.cpp" />
-    <ClCompile Include="src\main\core\plate_detect.cpp" />
-    <ClCompile Include="src\main\core\plate_judge.cpp" />
-    <ClCompile Include="src\main\core\plate_locate.cpp" />
-    <ClCompile Include="src\main\core\plate_recognize.cpp" />
-    <ClCompile Include="src\main\easypr\easypr_test.cpp" />
-    <ClCompile Include="src\main\train\ann_train.cpp" />
-    <ClCompile Include="src\main\train\svm_train.cpp" />
-    <ClCompile Include="src\main\util\util.cpp" />
-    <ClCompile Include="src\test\core\test_plate_recognize.cpp" />
+    <ClCompile Include="src\core\chars_identify.cpp" />
+    <ClCompile Include="src\core\chars_recognise.cpp" />
+    <ClCompile Include="src\core\chars_segment.cpp" />
+    <ClCompile Include="src\core\features.cpp" />
+    <ClCompile Include="src\experiment\hist_test.cpp" />
+    <ClCompile Include="src\train\svm_train.cpp" />
+    <ClCompile Include="src\util\mc_data_prepare.cpp" />
+    <ClCompile Include="src\experiment\hog_test.cpp" />
+    <ClCompile Include="src\main\main.cpp" />
+    <ClCompile Include="src\main\test.cpp" />
+    <ClCompile Include="src\core\plate_detect.cpp" />
+    <ClCompile Include="src\core\plate_judge.cpp" />
+    <ClCompile Include="src\core\plate_locate.cpp" />
+    <ClCompile Include="src\core\plate_recognize.cpp" />
+    <ClCompile Include="src\experiment\sift_test.cpp" />
+    <ClCompile Include="src\util\learn_prepare.cpp" />
+    <ClCompile Include="src\util\util.cpp" />
+    <ClCompile Include="src\experiment\ycbcr_test.cpp" />
   </ItemGroup>
   <ItemGroup>
-    <ClInclude Include="src\main\include\chars_identify.h" />
-    <ClInclude Include="src\main\include\chars_recognise.h" />
-    <ClInclude Include="src\main\include\chars_segment.h" />
-    <ClInclude Include="src\main\include\plate_detect.h" />
-    <ClInclude Include="src\main\include\plate_judge.h" />
-    <ClInclude Include="src\main\include\plate_locate.h" />
-    <ClInclude Include="src\main\include\plate_recognize.h" />
-    <ClInclude Include="src\main\include\prep.h" />
+    <ClInclude Include="src\include\chars_identify.h" />
+    <ClInclude Include="src\include\chars_recognise.h" />
+    <ClInclude Include="src\include\chars_segment.h" />
+    <ClInclude Include="src\include\features.h" />
+    <ClInclude Include="src\include\plate_detect.h" />
+    <ClInclude Include="src\include\plate_judge.h" />
+    <ClInclude Include="src\include\plate_locate.h" />
+    <ClInclude Include="src\include\plate_recognize.h" />
+    <ClInclude Include="src\include\prep.h" />
+    <ClInclude Include="src\include\util.h" />
   </ItemGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
   <ImportGroup Label="ExtensionTargets">

EasyPR.vcxproj.filters

@@ -24,78 +24,92 @@
     <None Include="关于版权.txt">
       <Filter>资源文件</Filter>
     </None>
-    <None Include="train\训练说明.txt" />
   </ItemGroup>
   <ItemGroup>
-    <ClCompile Include="src\main\easypr\data_prepare.cpp">
+    <ClCompile Include="src\main\main.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\easypr\easypr_main.cpp">
+    <ClCompile Include="src\experiment\ycbcr_test.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\easypr\data_proc.cpp">
+    <ClCompile Include="src\experiment\sift_test.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\easypr\easypr_test.cpp">
+    <ClCompile Include="src\experiment\hist_test.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\chars_identify.cpp">
+    <ClCompile Include="src\experiment\hog_test.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\chars_recognise.cpp">
+    <ClCompile Include="src\util\util.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\chars_segment.cpp">
+    <ClCompile Include="src\util\mc_data_prepare.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\plate_detect.cpp">
+    <ClCompile Include="src\util\learn_prepare.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\plate_judge.cpp">
+    <ClCompile Include="src\core\chars_identify.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\plate_locate.cpp">
+    <ClCompile Include="src\core\chars_recognise.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\core\plate_recognize.cpp">
+    <ClCompile Include="src\core\chars_segment.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\train\ann_train.cpp">
+    <ClCompile Include="src\core\plate_detect.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\train\svm_train.cpp">
+    <ClCompile Include="src\core\plate_judge.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\main\util\util.cpp">
+    <ClCompile Include="src\core\plate_locate.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
-    <ClCompile Include="src\test\core\test_plate_recognize.cpp">
+    <ClCompile Include="src\core\plate_recognize.cpp">
+      <Filter>源文件</Filter>
+    </ClCompile>
+    <ClCompile Include="src\core\features.cpp">
+      <Filter>源文件</Filter>
+    </ClCompile>
+    <ClCompile Include="src\main\test.cpp">
+      <Filter>源文件</Filter>
+    </ClCompile>
+    <ClCompile Include="src\train\svm_train.cpp">
       <Filter>源文件</Filter>
     </ClCompile>
   </ItemGroup>
   <ItemGroup>
-    <ClInclude Include="src\main\include\chars_identify.h">
+    <ClInclude Include="src\include\chars_identify.h">
+      <Filter>头文件</Filter>
+    </ClInclude>
+    <ClInclude Include="src\include\chars_recognise.h">
+      <Filter>头文件</Filter>
+    </ClInclude>
+    <ClInclude Include="src\include\chars_segment.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\chars_recognise.h">
+    <ClInclude Include="src\include\plate_detect.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\chars_segment.h">
+    <ClInclude Include="src\include\plate_judge.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\plate_detect.h">
+    <ClInclude Include="src\include\plate_locate.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\plate_judge.h">
+    <ClInclude Include="src\include\plate_recognize.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\plate_locate.h">
+    <ClInclude Include="src\include\prep.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\plate_recognize.h">
+    <ClInclude Include="src\include\util.h">
       <Filter>头文件</Filter>
     </ClInclude>
-    <ClInclude Include="src\main\include\prep.h">
+    <ClInclude Include="src\include\features.h">
       <Filter>头文件</Filter>
     </ClInclude>
   </ItemGroup>

README.md

@@ -1,60 +0,0 @@
-EasyPR
-======
-
-EasyPR是一个中文的开源车牌识别系统，其目标是成为一个简单、灵活、准确的车牌识别引擎。
-
-相比于其他的车牌识别系统，EasyPR有如下特点：
-
-* 它基于openCV这个开源库，这意味着所有它的代码都可以轻易的获取。
-* 它能够识别中文，例如车牌为苏EUK722的图片，它可以准确地输出std:string类型的"苏EUK722"的结果。
-* 它的识别率较高。目前情况下，字符识别已经可以达到90%以上的精度。
-
-### 例子
-
-假设我们有如下的原始图片，需要识别出中间的车牌字符与颜色：
-
-![EasyPR 原始图片](doc/res/plate_locate.jpg)
-
-经过EasyPR的第一步处理车牌检测（PlateDetect）以后，我们获得了原始图片中仅包含车牌的图块：
-
-![EasyPR 车牌](doc/res/blue_plate.jpg)
-
-接着，我们对图块进行OCR过程，在EasyPR中，叫做字符识别（CharsRecognize）。我们得到了一个包含车牌颜色与字符的字符串：
-
-“蓝牌：苏EUK722”
-
-
-### 安装
-
-EasyPR不需要安装，开发者直接在其上做改动。如果想使用DLL形式引用或者使用C#与Java调用，则可以在[EasyPR_DLL](https://github.com/liuruoze/EasyPR_Dll)中找到。
-
-详细的开发与教程请见[介绍与开发教程](http://www.cnblogs.com/subconscious/p/3979988.html)。
-
-### 使用
-
-使用Git克隆一份拷贝到你本机或者直接下载zip压缩吧。使用vs2010或以上版本的IDE打开sln文件后，就可以看到本项目的所有目录。
-
-以下表格是本工程中所有目录的解释与关系:
-
-|目录 | 解释
-|------|----------
-| src |  所有源文件
-| model | 机器学习的模型
-| train | 训练代码与数据
-| image | 测试用的图片
-| doc | 相关文档
-| tmp | 存放临时图片的目录
-
-以下表格是src目录中所有文件的解释与关系:
-
-|文件 | 解释
-|------|----------
-| plate_locate |  车牌定位
-| plate_judge | 车牌判断
-| plate_detect | 车牌检测，是车牌定位与车牌判断功能的组合
-| chars_segment | 字符分割
-| chars_identify | 字符鉴别
-| chars_recognise | 字符识别，是字符分割与字符鉴别功能的组合
-| plate_recognize | 车牌识别，是车牌检测与字符识别的共有子类
-| prep.h | 预包含头文件
-| test.cpp | 测试

doc/EasyPR介绍.txt

@@ -1,71 +0,0 @@
-很高兴能够在博客园开通博客，并发表这系列的文章。
-
-我下面要分享的这些内容主要是关于我正在做的一个开源系统---一个中文的车牌识别系统。
-
-我给它取的名字为EasyPR，也就是Easy to do Plate Recognition的意思。我开发这套系统的主要原因是因为我希望能够锻炼我在这方面的能力，包括C++技术、计算机图形学、机器学习等。我把这个项目开源的主要目的是：1.跟大家分享我的开发成果，其中包括我的训练数据集和模型。2.能够通过大家的智力的集合，可以让这套系统发展的更好，包括识别率更准确，鲁棒性更高等等。
-
-所以从一开始，EasyPR的目标就是奔着开源而来的，它生为开源项目，到它结束，也还是开源项目，这点不会有改变。
-
-相比于其他的车牌识别系统，EasyPR有如下特点：
-1.它基于openCV这个开源库，这意味着所有它的代码都可以轻易的获取。
-2.它能够识别中文，这是它的一个重要特点。
-3.它的识别率还可以，考虑到它还有很多待改进和优化的空间，这已经很不错了。
-
-那么，EasyPR是如何产生的呢？我简单介绍一下它诞生过程：
-
-首先，在5月份左右时我考虑要做一个车牌识别系统。这个车牌系统中所有的代码都应该是开源的，不能基于任何黑盒技术。这主要起源于我想锻炼自己的C++和计算机视觉的水平。
-
-我在网上开始搜索了资料。由于计算机视觉中很多的算法我都是使用openCV，而且openCV发展非常良好，因此我查找的项目必须得是基于OpenCV技术的。于是我在CSDN的博客上找了连续的两篇文章http://blog.csdn.net/jinshengtao/article/details/17883075 和http://blog.csdn.net/jinshengtao/article/details/17954427 。
-
-文章的作者taotao1233在这两篇博客中以半学习笔记半开发讲解的方式说明了一个车牌识别系统的全部开发过程。非常感谢他的这些博客，借助于这些资料，我着手开始了开发。当时的想法非常朴素，就是想看看按照这些资料，能否真的实现一个车牌识别的系统。关于车牌照片数据的问题，幸运的很，我正在开发的一个项目中有大量的照片，因此数据不是问题。
-
-令人高兴的是，系统确实能够工作，但是让人沮丧的，似乎也就“仅仅”能够工作而已。在车牌检测这个环节中正确性已经惨不忍睹。
-
-这个事情给了我一拨不小的冷水，本来我以为很快的开发进度看来是乐观过头了。于是我决定沉下心来，仔细研究他的系统实现的每一个过程，结合OpenCV的官网教程与API资料，我发现他的实现系统中有很多并不适合我目前在做的场景。
-
-我手里的数据大部分是高速上的图像抓拍数据，其中每个车牌都偏小，而且模糊度较差。直接使用他们的方法，正确率低到了可怕的地步。于是我开始尝试利用openCv中的一些函数与功能，替代，增加，调优等等方法，不断的优化。这个过程很漫长，但是也有很多的积累。我逐渐发现，并且了解他系统中每一个步骤的目的，原理以及如果修改可以进行优化的方法。
-
-在最终实现的代码中，我的代码已经跟他的原始代码有很多的不一样了，但是成功率大幅度上升，而且车牌的正确检测率不断被优化。在系列文章的后面，我会逐一分享这些优化的过程与心得。
-
-最终我实现的系统与他的系统有以下几点不同：
-1.他的系统代码基本上完全参照了《Mastering OpenCV with Practical Computer Vision Projects》这本书的代码，而这本书得代码是专门为西班牙车牌所开发的，因此并不适合中文的环境。
-2.他的系统的代码大部分是原始代码的搬迁，并没有做到优化与改进的地步。而我的系统中对原来的识别过程，做了很多优化步骤。
-3.车牌识别中核心的机器学习算法的模型，他直接使用了原书提供的，而我这两个过程的模型是自己生成，而且模型也做了测试，作为开源系统的一部分也提供了出来。
-
-尽管我和他的系统有这么多的不同，但是我们在根本的系统结构上是一致的。因该说，我们都是参照了“Mastering OpenCV”这本数的处理结构。在这点上，我并没有所“创新”，事实上，结果也证明了“Mastering OpenCV”上的车牌识别的处理逻辑，是一个实际有效的最佳处理流程。
-
-“Mastering OpenCV”，包括我们的系统，都是把车牌识别划分为了两个过程：即车牌检测（Plate Detection）和字符识别（Chars Recognition）两个过程。可能有些书籍或论文上不是这样叫的，但是我觉得，这样的叫法更容易理解，也不容易搞混。
-
-车牌检测（Plate Detection）：对一个包含车牌的图像进行分析，最终截取出只包含车牌的一个图块。这个步骤的主要目的是降低了在车牌识别过程中的计算量。如果直接对原始的图像进行车牌识别，会非常的慢，因此需要检测的过程。在本系统中，我们使用SVM（支持向量机）这个机器学习算法去判别截取的图块是否是真的“车牌”。
-
-字符识别（Chars Recognition）：有的书上也较Plate Recognition，我为了与整个系统的名称做区分，所以改叫做字符识别。这个步骤的主要目的就是从上一个车牌检测步骤中获取到的车牌图像，进行光学字符识别（OCR）这个过程。其中用到的机器学习算法是著名的人工神经网络（ANN）中的多层感知机（MLP）模型。最近一段时间非常火的“深度学习”其实就是多隐层的人工神经网络，与其有非常紧密的联系。通过了解光学字符识别（OCR）这个过程，也可以知晓深度学习所基于的人工神经网路技术的一些内容。
-
-下图是一个完整的EasyPR的处理流程：
-
-
-本开源项目的目标客户群有三类：
-1.需要开发一个车牌识别系统的（开发者）。
-2.需要车牌系统去识别车牌的（用户）。
-3.急于做毕业设计的（学生）。
-
-第一类客户是本项目的主要使用者，因此项目特地被精心划分为了6个模块，以供开发者按需选择。
-第二类客户可能会有部分，EasyPR有一个界面的应用程序，可以直接读取车牌。这个系统是通过WTL开发的，应该可以满足基本的使用需求。
-对于第三类客户，可以这么说，有完整的全套代码和详细的说明，我相信你们可以稍作修改就可以通过设计大考。
-
-当然，EasyUI的github地址也给在这里：
-
-推荐你使用EasyPR有以下几点理由：
-1.这里面的代码都是作者亲自优化过的，你可以在上面做修改，做优化，甚至一起协作开发，一些处理车牌的细节方法你应该是感兴趣的。
-2.如果你对代码不感兴趣，那么经过作者精心训练的模型，包括SVM和ANN的模型，可以帮助你提升或验证你程序的正确率。
-3.如果你对模型也不感兴趣，那么成百上千经过作者亲自挑选的训练数据生成的文件，你应该感兴趣。作者花了大量的时间处理这些训练数据与调整，现在直接提供给你，可以大幅度减轻很多人缺少数据的难题。
-
-在后面的文章组织中，我是这样计划的。花两篇文章介绍车牌检测和字符识别的内部细节，包括如何进行预处理以及机器学习算法如何训练的。接着用六篇左右的文章依次完整的介绍EasyPR中的所有模块，其中包括一些技术实现的思路等等内容。
-
-最后有一点，我需要说明的是：由于我本人工作的原因，我不可能把涉及到具体车牌的照片的数据提供出来，这会牵扯到犯法的问题。但是，通过EasyPR中机器学习生成的一些中间数据和最终模型，则是可以提供的。这些数据可以帮助那些拿不到太多训练数据的人们。如果有专门的训练数据的话，则可以按照作者文中的方法，自己去训练，效果会更好。
-
-
-
-
-
-
-

makefile.txt

@@ -1,20 +0,0 @@
-objects = main.o plate_locate.o plate_judge.o plate_detect.o \
-		chars_segment.o chars_identify.o chars_recognise.o plate_recognize.o
-
-cc = g++
-
-main : $(objects)
-	cc -o main $(objects)
-
-main.o : 
-plate_locate.o : plate_locate.h
-plate_judge.o : plate_judge.h
-plate_detect.o : plate_detect.h
-chars_segment.o : chars_segment.h
-chars_identify.o : chars_identify.h
-chars_recognise.o : chars_recognise.h
-plate_recognize.o : plate_recognize.h
-
-.PHONY : clean
-clean :
-	-rm edit $(objects)

src/core/features.cpp

@@ -0,0 +1,122 @@
+// 这个文件定义了EasyPR里所有特征生成的函数
+// 所属命名空间为easypr
+// 这个部分中的特征由easypr的开发者修改
+
+#include "../include/prep.h"
+#include "../include/features.h"
+
+/*! \namespace easypr
+    Namespace where all the C++ EasyPR functionality resides
+*/
+namespace easypr {
+
+//! 直方图均衡
+Mat histeq(Mat in)
+{
+	Mat out(in.size(), in.type());
+	if(in.channels()==3)
+	{
+		Mat hsv;
+		vector<Mat> hsvSplit;
+		cvtColor(in, hsv, CV_BGR2HSV);
+		split(hsv, hsvSplit);
+		equalizeHist(hsvSplit[2], hsvSplit[2]);
+		merge(hsvSplit, hsv);
+		cvtColor(hsv, out, CV_HSV2BGR);
+	}
+	else if(in.channels()==1)
+	{
+		equalizeHist(in, out);
+	}
+	return out;
+}
+
+// ！获取垂直和水平方向直方图
+Mat ProjectedHistogram(Mat img, int t)
+{
+	int sz=(t)?img.rows:img.cols;
+	Mat mhist=Mat::zeros(1,sz,CV_32F);
+
+	for(int j=0; j<sz; j++){
+		Mat data=(t)?img.row(j):img.col(j);
+		mhist.at<float>(j)=countNonZero(data);	//统计这一行或一列中，非零元素的个数，并保存到mhist中
+	}
+
+	//Normalize histogram
+	double min, max;
+	minMaxLoc(mhist, &min, &max);
+
+	if(max>0)
+		mhist.convertTo(mhist,-1 , 1.0f/max, 0);//用mhist直方图中的最大值，归一化直方图
+
+	return mhist;
+}
+
+
+//! 获得车牌的特征数
+Mat getTheFeatures(Mat in)
+{
+	const int VERTICAL = 0;
+	const int HORIZONTAL = 1;
+
+	//Histogram features
+	Mat vhist=ProjectedHistogram(in, VERTICAL);
+	Mat hhist=ProjectedHistogram(in, HORIZONTAL);
+
+	//Last 10 is the number of moments components
+	int numCols = vhist.cols + hhist.cols;
+
+	Mat out = Mat::zeros(1, numCols, CV_32F);
+
+	//Asign values to feature,样本特征为水平、垂直直方图
+	int j=0;
+	for(int i=0; i<vhist.cols; i++)
+	{
+		out.at<float>(j)=vhist.at<float>(i);
+		j++;
+	}
+	for(int i=0; i<hhist.cols; i++)
+	{
+		out.at<float>(j)=hhist.at<float>(i);
+		j++;
+	}
+
+	return out;
+}
+
+// ! EasyPR的getFeatures回调函数
+// ！本函数是生成直方图均衡特征的回调函数
+void getHisteqFeatures(const Mat& image, Mat& features)
+{
+	features = histeq(image);
+}
+
+// ! EasyPR的getFeatures回调函数
+// ！本函数是获取垂直和水平的直方图图值
+void getHistogramFeatures(const Mat& image, Mat& features)
+{
+	Mat grayImage;
+	cvtColor(image, grayImage, CV_RGB2GRAY);
+	Mat img_threshold;
+	threshold(grayImage, img_threshold, 0, 255, CV_THRESH_OTSU+CV_THRESH_BINARY);
+	features = getTheFeatures(img_threshold);
+}
+
+
+// ! EasyPR的getFeatures回调函数
+// ！本函数是获取SITF特征子
+// ! 
+void getSIFTFeatures(const Mat& image, Mat& features)
+{
+	//待完善
+}
+
+
+// ! EasyPR的getFeatures回调函数
+// ！本函数是获取HOG特征子
+void getHOGFeatures(const Mat& image, Mat& features)
+{
+	//待完善
+}
+
+}	/* \namespace easypr  */

src/core/plate_judge.cpp

@@ -1,4 +1,5 @@
 #include "../include/plate_judge.h"
+#include "../include/util.h"
 
 /*! \namespace easypr
     Namespace where all the C++ EasyPR functionality resides
@@ -9,6 +10,8 @@ CPlateJudge::CPlateJudge()
 {
 	//cout << "CPlateJudge" << endl;
 	m_path = "model/svm.xml";
+	m_getFeatures = getHistogramFeatures;
+
 	LoadModel();
 }
 
@@ -46,7 +49,27 @@ Mat CPlateJudge::histeq(Mat in)
 }
 
 
-//! 使用彩色图进行SVM判断
+//! 对单幅图像进行SVM判断
+int CPlateJudge::plateJudge(const Mat& inMat,int& result)
+{
+	if (m_getFeatures == NULL)
+		return -1;
+
+	Mat features;
+	m_getFeatures(inMat, features);
+
+	//通过直方图均衡化后的彩色图进行预测
+	Mat p = features.reshape(1, 1);
+	p.convertTo(p, CV_32FC1);
+
+	int response = (int)svm.predict(p);
+	result = response;
+
+	return 0;
+}
+
+
+//! 对多幅图像进行SVM判断
 int CPlateJudge::plateJudge(const vector<Mat>& inVec,
 								  vector<Mat>& resultVec)
 {
@@ -55,16 +78,14 @@ int CPlateJudge::plateJudge(const vector<Mat>& inVec,
 	{
 		Mat inMat = inVec[j];
 
-		//通过直方图均衡化后的彩色图进行预测
-		Mat p = histeq(inMat).reshape(1, 1);
-		p.convertTo(p, CV_32FC1);
-		int response = (int)svm.predict(p);
+		int response = -1;
+		plateJudge(inMat, response);
+
 		if (response == 1)
-		{
 			resultVec.push_back(inMat);
-		}
 	}
 	return 0;
 }
 
+
 }	/*! \namespace easypr*/