mmocr/demo/docs/ocr_demo.md

1.2 KiB

OCR End2End Demo


End-to-End Test Image Demo

To end-to-end test a single image with text detection and recognition simutaneously:

python demo/ocr_image_demo.py demo/demo_text_det.jpg demo/output.jpg
  • The default config for text detection and recognition are PSENet_ICDAR2015 and SAR, respectively.

  • The predicted result will be saved as demo/output.jpg.

  • To use other algorithms of text detection and recognition, please set arguments: --det-config, --det-ckpt, --recog-config, --recog-ckpt.

  • To use batch mode for text recognition, please set arguments: --batch-mode, --batch-size.

Remarks

  1. If --imshow is specified, the demo will also show the image with OpenCV.
  2. The ocr_image_demo.py script only supports GPU and so the --device parameter cannot take cpu as an argument.
  3. (Experimental) By specifying --ocr-in-lines, the ocr results will be grouped and presented in lines.