mmocr/demo
Sergio Bugallo Enjamio 18c54afbdc
Add support for batch inference (#86)
* Add support for numpy arrays in model_inference

* Add test for numpy ndarray inference

* Fix linting problems

* Add support for batch inference

* Add batch inference demo script

* Fix comment

* Test batch inference with paths and arrays

* lint code

* Update model_inference docstring

* Refactor model inference tests

* Change inference function to make text detectors and recognizers use the same input data types

* Change single state text detector model to support batch inference

* Lint code

* simplify inference tests

* Remove psenet from batch inference test cases to prevent the pytest being killed

* Update batch_image_demo.py

* fix bug when test with dataset 

fix bug when test with dataset, for example, `./tools/dist_test.sh configs/textrecog/sar/sar_r31_parallel_decoder_academic.py <checkpoint> 1 --eval acc`

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-12 02:04:01 +00:00
..
docs Hbsun/demo doc (#168) 2021-05-12 09:57:53 +08:00
README.md Hbsun/demo doc (#168) 2021-05-12 09:57:53 +08:00
batch_image_demo.py Add support for batch inference (#86) 2021-05-12 02:04:01 +00:00
demo_text_det.jpg [feature]: add textdet 2021-04-03 01:03:52 +08:00
demo_text_recog.jpg [feature]: add textdet 2021-04-03 01:03:52 +08:00
image_demo.py fix issue#127: changed save_path to out_file (#141) 2021-04-29 15:46:46 +08:00
ocr_image_demo.py use mmcv dump instead (#120) 2021-04-23 10:27:12 +08:00
webcam_demo.py Remove usage of \ (#49) 2021-04-06 12:16:46 +00:00

README.md

Demo

This page provides tutorials about running demos. Please click the link below for more information.

Please refer to Text Detection for the tutorial of Text Detection demo.




Please refer to Text Recognition for the tutorial of Text Recognition demo.




Please refer to End2End Demo for the tutorial of Text Detection and Recognition end-to-end demo.