Commit Graph

563 Commits (593d7529a31a2fad4aa789e776eafa5a4662e6c6)
 

Author SHA1 Message Date
lizz 0fd9ece7ea
[Fix] Fix kie demo (#610)
* Fix kie demo

Signed-off-by: lizz <lizz@sensetime.com>

* Update mmocr/utils/ocr.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/utils/ocr.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/utils/ocr.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-12-03 15:13:16 +08:00
Tong Gao e203d546b4
[Docs] Docstring for textdet models (#561)
* add docstring for dbnet
2021-12-02 18:57:01 +08:00
mpena-vina d3d56182c6
fix #630: Add input size check to model_inference (#633)
* fix #630: Add input size check to model_inference

* fix #630: Added exception when receiving empty image
2021-12-02 03:00:23 +00:00
Tong Gao 69d5040590
[Docs] Update model readme (#604)
* update satrn

* Add intro and pic to readmes

* update stats.py

* abstract for tps

* Fix CRNN and TPS CRNN readme
2021-11-30 16:40:18 +08:00
Hongbin Sun d1d2c1f469
[Fix] Change SBC case to DBC case (#632) 2021-11-30 15:17:07 +08:00
Tong Gao 1c8ab80409
[Enhancement] Pass custom_hooks to mmcv (#609) 2021-11-30 15:10:09 +08:00
Tong Gao efdc0e42fa
[Enhancement] Update CircleCI (#631) 2021-11-30 11:02:06 +08:00
Zaida Zhou 731a1b9195
[Enhancement] Support MMCV to 1.5.0 (#628) 2021-11-29 19:37:30 +08:00
Tong Gao 5654079839
[Fix] Fix DBNet_r50 config (#625) 2021-11-26 12:26:32 +00:00
Tong Gao 04ef6053a6
[Docs] Add MMFewShot (#621) 2021-11-26 11:18:57 +08:00
Hongbin Sun 3188e53404
[Refactor] refactor textrecog config structure (#617)
* refactor configs of textrecog

* remove duplicate key in config _base_

* fix typo

* rename dataset config file
2021-11-25 16:27:45 +08:00
Tong Gao 1544e9bb29
[Enhancement] Remove unnecessary custom_import from train.py (#603) 2021-11-25 15:59:35 +08:00
Tong Gao 809661ea28
[Enhancement] CircleCI Setup (#611)
* Add .circleci/config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* fix unit test

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml

* Updated config.yml
2021-11-23 19:54:38 +08:00
Tong Gao 53b050b0ca
[Docs] Add MMFlow & MIM (#597)
* [Docs] Add MMFlow

* add mim
2021-11-19 14:16:42 +08:00
Tong Gao e267d06281
[Fix] Always broadcast a random seed to all the processes (#600) 2021-11-18 22:26:21 +08:00
apiaccess21 284b5acc12
[Docs] C++ example section for deployment (#593)
* C++ example section

* Optimize the section for readthedocs

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-17 14:16:54 +08:00
Hongbin Sun 98c5bff1e0
[Feature] Save results to json file for kie. (#589)
* save results json for kie

* update config

* Fix KIE inference bug

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-15 20:28:13 +08:00
Hongbin Sun f2b9ba93be
[Fix] keep original texts for kie (#588) 2021-11-15 17:22:17 +08:00
Hongbin Sun 34e3ff40e1
[Fix] Fix the same start_point for different texts in draw_texts_by_pil (#587) 2021-11-15 17:19:09 +08:00
Tong Gao dc6d957967
[Feature] Avoid dupliate placeholder docs in CN (#582)
* [Feature] Avoid dupliate placeholder docs in CN

* add changelog to cn docs
2021-11-12 14:40:12 +08:00
Shouping Shan 18fde080c4
[Fix] Modify algorithm "sar" weights path in metafile (#581)
Original model `sar_r31_parallel_decoder_academic` weights path in sar/metafile.yml belongs to crnn which should be modified.
2021-11-12 12:55:56 +08:00
Hongbin Sun 4870ca91fc
[Fix] Fix resource warning: unclosed file (#577)
* fix resource warning: unclosed file

* fix import

* Eliminate det_test_img's dependency on mmdet

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-12 12:36:57 +08:00
Tong Gao c39f1160ac
add chinese ocr to ocr.py and fix its docs (#579) 2021-11-12 11:03:29 +08:00
Tong Gao a9fde9146f
[Feature] Improve dbnet r50 by fixing std (#578) 2021-11-11 19:10:12 +08:00
Tong Gao 51f2d9a4c3
[Docs] Docstring for text recognition models (#562) 2021-11-11 19:09:25 +08:00
Tong Gao 9de4ddf952
fix broken links (#576) 2021-11-11 17:57:15 +08:00
Tong Gao cae080734f
fix recog_test_imgs (#574) 2021-11-11 17:39:54 +08:00
Tong Gao 34bd6b9bbb
[Docs] Docs for KIE CloseSet & OpenSet (#573)
* add kie docs

* fix paths

* sync cn docs

* fix layout error
2021-11-11 17:39:27 +08:00
Hongbin Sun a50b0c9fb9
[Feature] Support openset kie (#498)
* add openset kie dataset

* updare readme

* add anno convert script

* update docstring

* update script

* add & update docstring

* fix typo

* update docstring format
2021-11-11 14:47:38 +08:00
Hongbin Sun 9f42d78db7
[Feature] Save filename in text detection test results (#570)
* save filename in textdet test results

* fix pytest
2021-11-11 14:07:07 +08:00
Tong Gao 37f8fcd841
[Fix] Always map location to cpu when loading checkpoint (#567) 2021-11-10 14:03:03 +08:00
Tong Gao fdae159b89
[Fix] Fix py3.9 CI (#563) 2021-11-10 14:02:33 +08:00
Tong Gao 1efdbda790
[Docs] Update docs menu (#564)
* update docs menu

* fix wrong links
2021-11-09 17:16:13 +08:00
Tong Gao 5fb1268a06
unify task names in metafiles (#548) 2021-11-03 15:03:37 +08:00
Tong Gao dab5725d70
fix paper field in metafiles (#550) 2021-11-03 15:02:31 +08:00
Darwin Bautista 80741e1479
[Feature] Add converter for the Open Images v5 text annotations by Krylov et al. (#497)
* Add converter for the OpenVINO annotations for Open Images by Krylov et al.

Open Images V5 Text Annotation and Yet Another Mask Text Spotter
Paper: https://arxiv.org/abs/2106.12326

* docs fix & add chinese docs
2021-10-28 16:49:36 +08:00
Tong Gao 2f983eca83
[Fix] Fix PyTorch 1.6 incompatible checkpoints (#540)
* fix pt1.6 incompatible checkpoints

* fix satrn metafile
2021-10-27 19:12:52 +08:00
Tong Gao 027cb7aac5
[Docs] TorchServe docs (#539)
* Add torchserve docs

* polish contents and fix the visiualization bug in torchserve testing

* polish content
2021-10-22 16:49:28 +08:00
liukuikun 5671939bbc
[Fix] fix converting to onnx bug because of changing key from img_shape to resize_shape (#523)
* fix pytorch to onnx for mmocr0.3.0

* fix pytorch to onnx for mmocr0.3.0

* modify deployment doc and fix scale_factor bug and test psenet and panet

* fix trt part

* init detector with **(cfg.model)

Co-authored-by: AllentDan <AllentDan@yeah.net>
2021-10-20 19:55:57 +08:00
Tong Gao 75d32504e0
[Feature] Add codespell pre-commit hook and fix typos (#520)
* add spellcheck hook

* fix typos
2021-10-13 22:56:11 +08:00
liukuikun 7d36a7e882
[Feature] Add torchserve support for text detection and recognition (#522)
* add torchserve

* add test torchserve

* delete unused postprocess and delete parse_results

* fix bug

* fix bug

* fix bug

* Update tools/deployment/test_torchserve.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update tools/deployment/mmocr_handler.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-10-13 22:21:16 +08:00
Tong Gao abfe75dc62
[Fix] Update checkpoint links for SATRN (#518) 2021-10-10 11:26:37 +08:00
fatfishZhao 66ce0efee0
[Docs] Add Chinese Translation of deployment.md. (#506)
* Add Chinese Translation of deployment.md.

* final fix

Co-authored-by: zhaoyuxuan <zhaoyuxuan@zhaoyuxuandeMacBook-Pro.local>
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-09-28 17:11:15 +08:00
Tong Gao 06c726c130
[Enhancement] Use container in CI (#502)
* use container
2021-09-28 14:05:10 +08:00
Tong Gao f72012092f
[Docs] Enhancement & sync Chinese docs (#492)
* reorder navigation links, add links to twitter and zhihu

* update chinese docs
2021-09-28 11:32:44 +08:00
Tong Gao aa4b5b5574
[Fix] Allow invalid polygons in intersection and union by default (#471)
* Allow invalid polygons in intersection and union by default

* update docs
2021-09-28 10:57:12 +08:00
Tong Gao 8c72d80164
[Fix] Fix image export in test.py for KIE models (#486)
* Fix image export in test.py for sdmgr model

* fix pretrained
2021-09-18 18:24:55 +08:00
EricTian 38bdc10f22
[Docs] Chinese docs for install.md (#364)
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-09-14 11:47:38 +08:00
Tong Gao e9a5350f47
[Enhancement] Update model-index.yml (#484) 2021-09-08 12:51:44 +08:00
Tong Gao de83b271eb
add py1.9 CI (#475) 2021-09-08 11:45:51 +08:00