Download-Tools/ccextractor
1
0
Fork 0
mirror of https://github.com/CCExtractor/ccextractor.git synced 2025-01-13 13:40:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add more tapping points while debugging tesseract

Browse Source 
When OCR_DEBUG is defined, code will dump more images
to find root cause of failed OCR
This commit is contained in:
Anshul Maheshwari 2018-12-22 14:12:53 +05:30
parent 9c20e0afb1
commit 0b76cc1991
2 changed files with 41 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/CHANGES.TXT
View File

@ -1,5 +1,6 @@
0.88(2018-10-24) (unreleased)
-----------------
- New: More tapping points for debug image in ccextractor.
- New: Add support for tesseract 4.0
- Optimize: Remove multiple RGB to grey conversion in OCR.
- Fix: Update UTF8Proc to 2.2.0

57
src/lib_ccx/ocr.c
View File

@ -239,17 +239,50 @@ BOX* ignore_alpha_at_edge(png_byte *alpha, unsigned char* indata, int w, int h,
return cropWindow;
}
void debug_tesseract(struct ocrCtx* ctx, char *dump_path)
{
#ifdef OCR_DEBUG
char str[1024] = "";
static int i = 0;
PIX* pix = NULL;
PIXA* pixa = NULL;
pix = TessBaseAPIGetInputImage(ctx->api);
sprintf(str,"%sinput_%d.jpg", dump_path, i);
pixWrite(str, pix, IFF_JFIF_JPEG);
pix = TessBaseAPIGetThresholdedImage(ctx->api);
sprintf(str,"%sthresholded_%d.jpg", dump_path, i);
pixWrite(str, pix, IFF_JFIF_JPEG);
TessBaseAPIGetRegions(ctx->api, &pixa);
sprintf(str,"%sregion_%d", dump_path, i);
pixaWriteFiles(str, pixa, IFF_JFIF_JPEG);
TessBaseAPIGetTextlines(ctx->api, &pixa, NULL);
sprintf(str,"%slines_%d", dump_path, i);
pixaWriteFiles(str, pixa, IFF_JFIF_JPEG);
TessBaseAPIGetWords(ctx->api, &pixa);
sprintf(str,"%swords_%d", dump_path, i);
pixaWriteFiles(str, pixa, IFF_JFIF_JPEG);
i++;
#endif
}
char* ocr_bitmap(void* arg, png_color *palette,png_byte *alpha, unsigned char* indata,int w, int h, struct image_copy *copy)
{
// uncomment the below lines to output raw image as debug.png iteratively
// save_spupng("debug.png", indata, w, h, palette, alpha, 16);
PIX *pix = NULL;
PIX *cpix = NULL;
PIX *cpix_gs = NULL; // Grayscale version
PIX *color_pix = NULL;
PIX *color_pix_out = NULL;
char*text_out= NULL;
PIX* pix = NULL;
PIX* cpix = NULL;
PIX* cpix_gs = NULL; // Grayscale version
PIX* color_pix = NULL;
PIX* color_pix_out = NULL;
char* text_out = NULL;
int i,j,index;
unsigned int wpl;
unsigned int *data,*ppixel;
@ -298,24 +331,14 @@ char* ocr_bitmap(void* arg, png_color *palette,png_byte *alpha, unsigned char* i
BOX *crop_points = ignore_alpha_at_edge(copy->alpha, copy->data, w, h, color_pix, &color_pix_out);
// Converting image to grayscale for OCR to avoid issues with transparency
cpix_gs = pixConvertRGBToGray(cpix, 0.0, 0.0, 0.0);
#ifdef OCR_DEBUG
{
char str[128] = "";
static int i = 0;
sprintf(str,"temp/file_c_%d.jpg",i);
printf("Writing file_c_%d.jpg\n", i);
pixWrite(str, cpix_gs, IFF_JFIF_JPEG);
i++;
}
#endif
if (cpix_gs==NULL)
tess_ret=-1;
else
{
TessBaseAPISetImage2(ctx->api, cpix_gs);
color_pix_out = TessBaseAPIGetThresholdedImage(ctx->api);
tess_ret = TessBaseAPIRecognize(ctx->api, NULL);
debug_tesseract(ctx, "./temp/");
if (tess_ret) {
mprint("\nIn ocr_bitmap: Failed to perform OCR. Skipped.\n");