mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmocr/docs/text_recog_demo.md

1.7 KiB
Raw Blame History

Text Recognition Demo

Text Recognition Image Demo

We provide a demo script to test a single demo image for text recognition with a single GPU.

Text Recognition Model Preparation: The pre-trained text recognition model can be downloaded from model zoo. Take SAR as an example:

python demo/image_demo.py ${TEST_IMG} ${CONFIG_FILE} ${CHECKPOINT_FILE} ${SAVE_PATH} [--imshow] [--device ${GPU_ID}]

Example:

python demo/image_demo.py demo/demo_text_recog.jpg configs/textrecog/sar/sar_r31_parallel_decoder_academic.py https://download.openmmlab.com/mmocr/textrecog/sar/sar_r31_parallel_decoder_academic-dba3a4a3.pth demo/demo_text_recog_pred.jpg

The predicted result will be saved as demo/demo_text_recog_pred.jpg.

Text Recognition Webcam Demo

We also provide live demos from a webcam as in mmdetection.

python demo/webcam_demo.py \
    ${CONFIG_FILE} \
    ${CHECKPOINT_FILE} \
    [--device ${GPU_ID}] \
    [--camera-id ${CAMERA-ID}] \
    [--score-thr ${SCORE_THR}]

Examples:

python demo/webcam_demo.py \
    configs/textrecog/sar/sar_r31_parallel_decoder_academic.py \
    https://download.openmmlab.com/mmocr/textrecog/sar/sar_r31_parallel_decoder_academic-dba3a4a3.pth

Remarks

  1. If --imshow is specified, the demo will also show the image with OpenCV.
  2. The image_demo.py script only supports GPU and so the --device parameter cannot take cpu as an argument.