Android NDK开发:实战案例-电动车牌号识别(图像的处理和识别)

目录

图像二值化处理

这里主要是将图像处理成只有黑白两种颜色的图像,这样有利于图像的识别,这里为了提升处理的速度因此我选择NDK层来进行图像的处理。
首先需要在CMakeLists.txt中连接jnigraphics库:



然后在native-lib.cpp中编写处理图像的方法:

#include 
#include 
#include 
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展示,效果如下:


案例源码

https://gitee.com/itfitness/LicenseOcrDemo

你可能感兴趣的:(Android NDK开发:实战案例-电动车牌号识别(图像的处理和识别))