mmocr

2.7 KiB

Raw Blame History

Text Recognition Demo

Text Recognition Single 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 Multiple Image Demo

We provide a demo script to test multi-images in batch mode 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/batch_image_demo.py ${CONFIG_FILE} ${CHECKPOINT_FILE} ${SAVE_PATH} --images ${IMAGE1} ${IMAGE2} [--imshow] [--device ${GPU_ID}]

Example:

python demo/image_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 save_results --images demo/demo_text_recog.jpg demo/demo_text_recog.jpg

The predicted result will be saved in folder save_results.

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

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

2.7 KiB Raw Blame History