目录
图像二值化处理
这里主要是将图像处理成只有黑白两种颜色的图像,这样有利于图像的识别,这里为了提升处理的速度因此我选择NDK层来进行图像的处理。
首先需要在CMakeLists.txt中连接jnigraphics库:
然后在native-lib.cpp中编写处理图像的方法:
#include <jni.h>
#include <string>
#include <android/bitmap.h>
extern "C"
JNIEXPORT void JNICALL
Java_com_itfitness_licenseocrdemo_utils_CardUtil_binaryBitmap(JNIEnv *env, jclass type,
jobject jBitmap) {
int result;
// 获取源Bitmap相关信息:宽、高等
AndroidBitmapInfo sourceInfo;
result = AndroidBitmap_getInfo(env, jBitmap, &sourceInfo);
if (result < 0) {
return;
}
// 获取源Bitmap像素数据 这里用的是32位的int类型 argb每个8位
uint32_t* sourceData;
//锁定像素的地址(不锁定的话地址可能会发生改变)
result = AndroidBitmap_lockPixels(env, jBitmap, (void**)& sourceData);
if (result < 0) {
return;
}
// 遍历各个像素点
int color;
int red, green, blue , alpha;
int width = sourceInfo.width;
int height = sourceInfo.height;
int w, h;
for (h = 0; h < height; h++) {
for (w = 0; w < width; w++) {
color = sourceData[h * width + w];
alpha = color & 0xff000000;
red = (color & 0x00ff0000) >> 16;
green = (color & 0x0000ff00) >> 8;
blue = color & 0x000000ff;
// 通过加权平均算法,计算出最佳像素值
color = red * 0.3 + green * 0.59 + blue * 0.11;
if (color <= 180) {
color = 0;
}
else
{
color = 255;
}
sourceData[h * width + w] = alpha | (color << 16) | (color << 8) | color;
}
}
AndroidBitmap_unlockPixels(env, jBitmap);
}
当然图像的处理也可以用Java代码实现,具体可以看我的这篇文章中的案例:Android NDK开发:操作Bitmap实现图像二值化
处理后的扫描图像如下:
图像的识别
图像的识别使用的是tess-two(OCR库)
//OCR
implementation 'com.rmtheis:tess-two:9.0.0'
这里还需要一个用于识别的文件,这里先用我项目中的,在这个系列下一次的文章我将告诉大家如何自己生成识别文件
然后我们需要将该文件拷贝到SD卡中,其中FileUtil是我写的一个工具类,在文章末尾的案例源码中可以查看:
private void initTessTwo() {
new Thread(new Runnable() {
@Override
public void run() {
try {
FileUtil.copyRes2SD(ScanningCameraView.TESSDATA,ScanningCardOcrActivity.this);
}catch (Exception e){
}
}
}).start();
}
对图像识别的代码如下:
//文字识别
TessBaseAPI baseApi = new TessBaseAPI();
//初始化OCR的字体数据,DATA_PATH为路径,DEFAULT_LANGUAGE指明要用的字体库(不用加后缀)
baseApi.init(DATA_PATH, "num_1");//这里名字要和raw中的文件名一样
//设置识别模式
baseApi.setPageSegMode(TessBaseAPI.PageSegMode.PSM_AUTO);
//设置要识别的图片
baseApi.setImage(cardNumberBitmap);
//开始识别
String result = baseApi.getUTF8Text();
baseApi.clear();
baseApi.end();
最后将识别后的车牌号返回MainActivity展示,效果如下:
网友评论