mmocr

Commit Graph

Author	SHA1	Message	Date
Hongbin Sun	82f64a5b62	support batch inference during testing (#310 ) * support batch inference during testing * fix unittest * update docs using url * set cfg for train, val and test * update docs * update docs and test.py * samples_per_gpu as global setting * changes revert	2021-06-23 11:34:29 +08:00
AllentDan	d57f279083	[Fix] Correct isort setup (#288 ) * isort * remove tools from setup.cfg	2021-06-16 19:44:01 +08:00
Hongbin Sun	97e2f27017	fix #173 : support aug test (#178 ) * fix #173: support aug test * fix pytest * support batch inference for both text det and recog * update unittest * use one img for batch infer test	2021-05-13 07:18:00 +00:00
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
Hongbin Sun	43dcb32d4f	Hbsun/end2end demo (#105 ) * add end2end demo * fix typo * pad box * fix bug of crnn * fix polygon * update docstring * fix bug of polygon * updare demo api * fix except * rename * fix with comments	2021-04-22 20:42:42 +08:00
Hongbin Sun	f5c385b02c	fix inference bug (#87 ) * fix inference bug * add unit test for det	2021-04-19 02:07:25 +00:00
Sergio Bugallo Enjamio	a68de2214a	Add support for numpy array inference (#74 ) * Add support for numpy arrays in model_inference * Add test for numpy ndarray inference * Fix linting problems * Simplify assertion in model_inference	2021-04-18 20:10:17 +08:00
lizz	44ca9c2a61	Remove usage of \ (#49 ) * Remove usage of \ Signed-off-by: lizz <lizz@sensetime.com> * rebase Signed-off-by: lizz <lizz@sensetime.com> * typos Signed-off-by: lizz <lizz@sensetime.com> * Remove test dependency on tools/ Signed-off-by: lizz <lizz@sensetime.com> * Remove usage of \ Signed-off-by: lizz <lizz@sensetime.com> * rebase Signed-off-by: lizz <lizz@sensetime.com> * typos Signed-off-by: lizz <lizz@sensetime.com> * Remove test dependency on tools/ Signed-off-by: lizz <lizz@sensetime.com> * typo Signed-off-by: lizz <lizz@sensetime.com> * KIE in keywords Signed-off-by: lizz <lizz@sensetime.com> * some renames Signed-off-by: lizz <lizz@sensetime.com> * kill isort skip Signed-off-by: lizz <lizz@sensetime.com> * aggregation discrimination Signed-off-by: lizz <lizz@sensetime.com> * aggregation discrimination Signed-off-by: lizz <lizz@sensetime.com> * tiny Signed-off-by: lizz <lizz@sensetime.com> * fix bug: model infer on cpu Co-authored-by: Hongbin Sun <hongbin306@gmail.com>	2021-04-06 12:16:46 +00:00
Hongbin Sun	03a270d2c2	fix #47 : add unit test for api (#48 )	2021-04-06 08:04:14 +00:00

9 Commits (c1ae3a41e5fafc0e64bece2be108f1867214cd26)