Android使用Tesseract-ocr进行文字识别

Vaccae

发布于 2019-07-25 14:25:17

9K0

发布于 2019-07-25 14:25:17

前言

Tessseract是一款由HP实验室开发由 Google 维护的开源 OCR（Optical Character Recognition , 光学字符识别）引擎。能够支持中文十分难得。虽然其识别效果不是很理想，但是对于要求不高的中小型项目来说，已经足够用了。

Tesseract-OCR下载地址

文字识别一般都用的tesseract-ocr。 GitHub：https://github.com/tesseract-ocr/tesseract

我们今天在Android上应用推荐的有个tess-two

GitHub：https://github.com/rmtheis/tess-two

还有一个字体识别库Tessdata（chi_sim.traineddata中文简体，chi_tra.traineddata中文繁体，eng.traineddata 英文库）

GitHub: https://github.com/tesseract-ocr/tessdata

演示效果

上图中，整张图进行识别，我感觉效果还可以，如果前置用OPENCV做图像的预处理后，可能效果会更好。

代码实现

首先下载tess-two和字体库

下载完成到我们的目录中

新建一个TesserartDemo的项目，导入tess-two

建好项目后，我们通过Import Module导入tess-two，由于tess-two是个ndk的项目，所以我们必须要在SDK Tools里面加入CMake和NDK，如下图，具体NDK可以看看以前的文章《Android NDK编程（一）---NDK介绍及环境搭建》

，

导入完tess-two后，我们进行编译，结果发现编译不过去，提示android-maven的错误。网上找了找资料，发现了解决办法。在tess-two的build.gradle的文件中配置

buildscript {
    repositories {
        jcenter()
    }
    dependencies {
        classpath 'com.android.tools.build:gradle:2.1.2'
        classpath 'org.codehaus.groovy:groovy-backports-compat23:2.3.5'
        classpath 'com.jfrog.bintray.gradle:gradle-bintray-plugin:1.0'
        classpath 'com.github.dcendents:android-maven-gradle-plugin:1.5'
    }
}

apply plugin: 'com.android.library'
apply plugin: 'com.github.dcendents.android-maven'
apply plugin: 'com.jfrog.bintray'

如下图：

重新编译，这次时间较长，我自己的电话大概花了十几分钟才编译完成。

拷贝字库文件进Android设备

我们利用Device File Explorer打开我们的虚拟机，将chi_sim.traineddata的字库文件拷贝到mnt/sdcard/tesserart/tessdata下，没有这个目录可以自己创建一下，后面会说到为什么拷贝到这里

AndroidManifest.xml

在这里面加入读取内存卡和写入内存卡的权限

    <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE" />
    <uses-permission android:name="android.permission.READ_PHONE_STATE" />

在res/drawable里面把我们的图片文件复制进来

activity_main.xml布局文件

在布局文件中加入一个imageview，一个button和一个TextView

<?xml version="1.0" encoding="utf-8"?>
<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"
    xmlns:app="http://schemas.android.com/apk/res-auto"
    xmlns:tools="http://schemas.android.com/tools"
    android:orientation="vertical"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    tools:context=".MainActivity">

    <ImageView
        android:layout_width="match_parent"
        android:layout_height="300dp"
        android:id="@+id/imgv"/>

    <Button
        android:layout_width="wrap_content"
        android:layout_height="wrap_content"
        android:id="@+id/btnOcr"
        android:text="识别"/>

    <TextView
        android:layout_width="wrap_content"
        android:layout_height="wrap_content"
        android:text="Hello World!"
        android:id="@+id/tvshow" />

</LinearLayout>

MainActivity.java

public class MainActivity extends AppCompatActivity {

    private ImageView imgv;
    private TextView tvshow;
    private Button btnocr;
    private Bitmap bmp;

    //如果Android的版本大于23，路径取根目录下的tesserart，小于的话是
    //在mnt/sdcard下面
    private String DATAPATH=Environment.getExternalStorageDirectory().getAbsolutePath() +
            File.separator + "tesserart";

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        requestPermission();


        imgv=findViewById(R.id.imgv);
        //获取自己的图片，这里是自己放在项目里面的。
        bmp=BitmapFactory.decodeResource(this.getResources(), R.drawable.tiddmg);
        imgv.setImageBitmap(bmp);
        imgv.setScaleType(ImageView.ScaleType.FIT_XY);

        tvshow=findViewById(R.id.tvshow);
        btnocr=findViewById(R.id.btnOcr);

        btnocr.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View view) {

                try {
                    TessBaseAPI mTess=new TessBaseAPI();
                    //存放tessdata的文件路径 就是chi_sim.traineddata文件的位置chi_sim.traineddata
                    //选择语言 chi_sim 简体中文  eng 英文
                    String language="chi_sim";

                    File dirdatapath=new File(DATAPATH);
                    if (!dirdatapath.exists()) {
                        dirdatapath.mkdirs();
                    }
                    File dir=new File(DATAPATH,"tessdata");
                    if (!dir.exists())
                        dir.mkdirs();
                    tvshow.append(DATAPATH+"\r\n");
                    mTess.init(DATAPATH, language);
                    //将图片设置到mTess进行识别
                    mTess.setImage(bmp);
                    //获取识别的文字（这里会等一段时间，这里的代码是在主线程的，建议将这部分代码放到子线程）
                    String result=mTess.getUTF8Text();
                    tvshow.append("结果为:" + result);
                } catch (Exception e) {
                    e.printStackTrace();
                    tvshow.append(e.getMessage());
                }
            }
        });

    }

    void requestPermission() {
        final int REQUEST_CODE=1;
        if (ContextCompat.checkSelfPermission(this, Manifest.permission.CAMERA) != PackageManager.PERMISSION_GRANTED) {
            ActivityCompat.requestPermissions(this, new String[]{
                            Manifest.permission.WRITE_EXTERNAL_STORAGE,
                            Manifest.permission.READ_EXTERNAL_STORAGE},
                    REQUEST_CODE);
        }
    }

}

重点说明：

上图中DATAPATH的文件路径，用Environment.getExternalStorageDirectory().getAbsolutePath()，在API23以下的时候，路径就是我们拷贝的mnt\sdcard\，如果在API23或以上直接就是我们的根目录，我自己的模拟器是Android 4.1的也就是API18，所以这里直接拷文件拷到mnt\sdcard下了。刚才测试时用了自己手机（Android9.0)，发现在mnt的sdcard下也有，可能我自己研究错了这里，先划掉吧。