RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

avfilter/vf_ocr: also export confidence of result

This commit is contained in:
Paul B Mahol 2019-05-17 19:00:47 +02:00
parent 3f31726994
commit 96c79d4e1f
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
doc/filters.texi
View File

@ -12910,6 +12910,7 @@ Set character blacklist.
@end table
The filter exports recognized text as the frame metadata @code{lavfi.ocr.text}.
The filter exports confidence of recognized words as the frame metadata @code{lavfi.ocr.confidence}.
@section ocv

10
libavfilter/vf_ocr.c
View File

@ -100,11 +100,21 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
AVFilterLink *outlink = ctx->outputs[0];
OCRContext *s = ctx->priv;
char *result;
int *confs;
result = TessBaseAPIRect(s->tess, in->data[0], 1,
in->linesize[0], 0, 0, in->width, in->height);
confs = TessBaseAPIAllWordConfidences(s->tess);
av_dict_set(metadata, "lavfi.ocr.text", result, 0);
for (int i = 0; confs[i] != -1; i++) {
char number[256];
snprintf(number, sizeof(number), "%d ", confs[i]);
av_dict_set(metadata, "lavfi.ocr.confidence", number, AV_DICT_APPEND);
}
TessDeleteText(result);
TessDeleteIntArray(confs);
return ff_filter_frame(outlink, in);
}