android - recognized text from tess-two library is wrong

醉酒当歌 提交于 2019-12-23 01:28:11

问题


I am trying to use the tess-two library to recognize text from imagae.

Here is my code:

load.setOnClickListener(new View.OnClickListener() {

            @Override
            public void onClick(View v) {
                // recognize text
                Bitmap temp = loadJustTakenImage(); //loads taken image from sdcard
                Bitmap rotatedImage = rotateIfNeeded(temp); // rotate method i found in some tutorial
                String text1 = recognizeText(rotatedImage);
            }

        });

Recognize text method:

(tessdata folder is in Download with the eng.traineddata and other files)

private String recognizeText(Bitmap bitmap) {
        // TODO Auto-generated method stub
        TessBaseAPI baseApi = new TessBaseAPI();
        // DATA_PATH = Path to the storage
        // lang = for which the language data exists, usually "eng"
        baseApi.init(Environment.getExternalStorageDirectory().toString()
                + "/Download/", "eng");
        // Eg. baseApi.init("/mnt/sdcard/tesseract/tessdata/eng.traineddata",
        // "eng");
        baseApi.setImage(bitmap);
        String recognizedText = baseApi.getUTF8Text();
        baseApi.end();
        return recognizedText;
    }

rotate image method:

private Bitmap rotateIfNeeded(Bitmap bitmap) {
        ExifInterface exif = null;
        try {
            exif = new ExifInterface(directoryPath+"/"+currentFileName+".jpg");
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        int exifOrientation = exif
                .getAttributeInt(ExifInterface.TAG_ORIENTATION,
                        ExifInterface.ORIENTATION_NORMAL);

        int rotate = 0;

        switch (exifOrientation) {
        case ExifInterface.ORIENTATION_ROTATE_90:
            rotate = 90;
            break;
        case ExifInterface.ORIENTATION_ROTATE_180:
            rotate = 180;
            break;
        case ExifInterface.ORIENTATION_ROTATE_270:
            rotate = 270;
            break;
        }

        if (rotate != 0) {
            int w = bitmap.getWidth();
            int h = bitmap.getHeight();

            // Setting pre rotate
            Matrix mtx = new Matrix();
            mtx.preRotate(rotate);

            // Rotating Bitmap & convert to ARGB_8888, required by tess
            bitmap = Bitmap.createBitmap(bitmap, 0, 0, w, h, mtx, false);
        }
        bitmap = bitmap.copy(Bitmap.Config.ARGB_8888, true);
        return bitmap;
    }

the text I am getting is a real mess, for example

for this image:

I got this text:

,7‘

Sometimes I just get an empty String.

What am I doing wrong?


回答1:


Based on the type of image on which you are trying to detect the characters, setting an appropriate Page segmentation mode will help detect the characters.

For example :

baseAPI.setPageSegMode(TessBaseAPI.PageSegMode.PSM_AUTO_ONLY);

The various other Page segmentation values are present in TessBaseApi.java :

/** Page segmentation mode. */
public static final class PageSegMode {
    /** Orientation and script detection only. */
    public static final int PSM_OSD_ONLY = 0;

    /** Automatic page segmentation with orientation and script detection. (OSD) */
    public static final int PSM_AUTO_OSD = 1;

    /** Fully automatic page segmentation, but no OSD, or OCR. */
    public static final int PSM_AUTO_ONLY = 2;

    /** Fully automatic page segmentation, but no OSD. */
    public static final int PSM_AUTO = 3;

    /** Assume a single column of text of variable sizes. */
    public static final int PSM_SINGLE_COLUMN = 4;

    /** Assume a single uniform block of vertically aligned text. */
    public static final int PSM_SINGLE_BLOCK_VERT_TEXT = 5;

    /** Assume a single uniform block of text. (Default.) */
    public static final int PSM_SINGLE_BLOCK = 6;

    /** Treat the image as a single text line. */
    public static final int PSM_SINGLE_LINE = 7;

    /** Treat the image as a single word. */
    public static final int PSM_SINGLE_WORD = 8;

    /** Treat the image as a single word in a circle. */
    public static final int PSM_CIRCLE_WORD = 9;

    /** Treat the image as a single character. */
    public static final int PSM_SINGLE_CHAR = 10;

    /** Find as much text as possible in no particular order. */
    public static final int PSM_SPARSE_TEXT = 11;

    /** Sparse text with orientation and script detection. */
    public static final int PSM_SPARSE_TEXT_OSD = 12;

    /** Number of enum entries. */
    public static final int PSM_COUNT = 13;
}

You can experiment with different page segmentation enum values and see which gives the best result.

For the above image, it seems like setting page segmentation to 'PSM_SINGLE_LINE' should yield the result you are looking for.



来源:https://stackoverflow.com/questions/29984673/android-recognized-text-from-tess-two-library-is-wrong

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!