62 Commits

Author SHA1 Message Date
jiangqing.vendor
27261b2bce [TODO] mv configs/base/models -> configs/textdet & textrecog 2022-07-21 10:58:03 +08:00
xinyu
567aec5390 update metafiles 2022-07-21 10:58:03 +08:00
jiangqing.vendor
dc84187311 [TODO] Replace loss_module with module_loss 2022-07-21 10:57:31 +08:00
jiangqing.vendor
3709c7b03a [Metafiles] Update Metafiles for FCENet, MaskRCNN, NRTR, MASTER 2022-07-21 10:57:31 +08:00
gaotongxiao
914c8af7bf Revert "[TODO] Add LoadImageFromLMDB"
This reverts commit e716ae726f007f79effdf2d45b4955a887f3c1e3
2022-07-21 10:57:30 +08:00
jiangqing.vendor
19958fbf6f [TODO] Add LoadImageFromLMDB 2022-07-21 10:57:17 +08:00
gaotongxiao
2d478ea244 Add ABINet cfg 2022-07-21 10:55:46 +08:00
liukuikun
77ffe8fb00 [Config] scanner config 2022-07-21 10:55:46 +08:00
liukuikun
ca35c78e69 [Config] SAR config 2022-07-21 10:51:03 +08:00
jiangqing.vendor
b406f3785f [MASTER] Add MASTER config 2022-07-21 10:51:02 +08:00
wangxinyu
3992f0d78e [SATRN] SATRN Config 2022-07-21 10:51:01 +08:00
jiangqing.vendor
4f2ec6de71 [NRTR] NRTR config 2022-07-21 10:51:01 +08:00
liukuikun
f1eebe9e34 [CRNN] CRNN config 2022-07-21 10:51:00 +08:00
liukuikun
e1e26d3f74
[Enchance] add codespell ignore and use mdformat (#1022)
* update

* update contributing

* update ci

* fix md

* update pre-commit hook

* update mdformat

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-06-09 14:58:44 +08:00
Qing Jiang
0c060099e1
[Fix] Fix config name of MASTER in ocr.py (#1044)
* loss

* fix

* 'fix'

* [update] full fix
2022-05-28 20:47:00 +08:00
Jianquan Ye
061d353b2a
[Model] Add MASTER (#807)
* fix #794: add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* Fix linting

* after git rebase main

* after git rebase main

* fix conflict add MASTER

* fix conflict add MASTER

* after git rebase main

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* after git rebase main

* add GCAModule to plugins

* coexist master and master_old

* fix merge mmocr 0.5.0 conflict

* fix lint error

* update

* [fix] remove remains in __init__

* [update] update code in review

* update readme for master

* Add docstr to MasterDecoder, refined MasterDecoder, remove MASTERLoss

* Unify the output length of MasterDecoder in train and test mode; add test for it, remove MasterLoss

* update readme

* update

* update metafile,README,demo/README,config,ocr.py

* Update mmocr/utils/ocr.py

* update

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
Co-authored-by: Mountchicken <mountchicken@outlook.com>
2022-05-05 16:06:15 +08:00
Tong Gao
d901b37035
[Fix] Add num_classes to configs of ABINet (#805)
* [Fix] Add num_classes to configs of ABINet

* add comments
2022-03-16 17:17:02 +08:00
liukuikun
976b976e4a
[Docs] update readme according to standard (#742)
* config/readme standard

* add ABINet linking

* fix linking error, delete unused cite and adjust note sytle

* union note format

* Remove > before paper link when generating docs
2022-01-25 23:02:59 +08:00
Hongbin Sun
5176e1bd5a
[Fix] fix link (#726) 2022-01-12 10:25:07 +08:00
Tong Gao
9104667112
[Model] Full ABINet Framework (#651)
Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2021-12-15 11:21:54 +08:00
Hongbin Sun
0a1787d6bc
[Refactor] refactor transformer modules (#618)
* base refactor

* update config

* modify implementation of nrtr

* add config file

* add mask

* add operation order

* fix contiguous bug

* update config

* fix pytest

* fix pytest

* update readme

* update readme and metafile

* update docstring

* fix norm cfg and dict size

* rm useless

* use mmocr builder instead

* update pytest

* update

* remove useless

* fix ckpt name

* fix path

* include all config file into pytest

* update inference

* Update test_recog_config.py
2021-12-04 17:12:31 +08: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
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
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
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
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
abfe75dc62
[Fix] Update checkpoint links for SATRN (#518) 2021-10-10 11:26:37 +08:00
Tong Gao
4c43f3fc47
[Docs] Fix a model link and add the metafile for SATRN (#473)
* update model link

* add metafile
2021-09-01 03:01:11 +00:00
LvTianlei
8f377f2fa2
[Feature] Add Satrn (#405)
* Add SATRN

* Create satrn_small_academic.py

* Update README.md

* change config name

* Update mmocr/models/textrecog/backbones/shallow_cnn.py

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

* Update configs/textrecog/satrn/satrn_academic.py

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

* Update configs/textrecog/satrn/satrn_small.py

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

* Update shallow_cnn.py

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update test_ocr_encoder.py

* change keep_aspect_ratio=False

* Update transformer_layer.py

* Update configs/textrecog/satrn/satrn_small.py

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

* Update configs/textrecog/satrn/satrn_academic.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update mmocr/models/textrecog/layers/transformer_layer.py

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

* Update transformer_layer.py

* Apply suggestions from code review

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

* Update transformer_layer.py

* update satrn readme

* add satrn to ocr.py

* add satrn_sm and fix configs

* add a test for config

* add copyright info

* use mmocr registry

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-08-19 22:02:58 +08:00
Hongbin Sun
3707d67106
support batch inference for crnn and segocr (#407)
* support batch inference for crnn and segocr
2021-08-03 16:52:58 +08:00
Hongbin Sun
6d97d8286e
Update nrtr batch size (#370) 2021-07-16 15:06:36 +08:00
Tong Gao
1a5a880abb
fix nrtr cfg (#356) 2021-07-08 13:30:58 +08:00
Hongbin Sun
20f65fb521
use unifor_concat_datase instead of concat_dataset (#343)
* use unifor_concat_datase instead of concat_dataset

* add unittest for uniform_concat_dataset
2021-07-03 16:03:37 +08:00
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
Hongbin Sun
1240169f18
update metafile (#183) 2021-05-14 00:17:33 +08:00
quincylin1
7e0d48df11
fix metafiles (#182) 2021-05-13 17:28:59 +08:00
quincylin1
cc27d8254b
Add metafiles (#179)
* add metafiles

* add metafiles

* add metafiles

* fix metafile errors

* fix metafiles

* Update metafile.yml

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-13 02:46:26 +00:00
Hongbin Sun
df5493a79e
refactor tps config (#135)
* refactor tps config

* recover tps

* add ckpt of tps
2021-05-12 06:14:24 +00:00
kang sheng
e7f27ae317
change argument names according to convention (#131)
* change argument names according to convention

* bug fix when rename leakyRelu
2021-05-11 16:07:36 +08:00
Hongbin Sun
892d486d01
add model for chinese (#156)
* add model for chinese

* update readme

* update readme

* fix link error; add vis for Chinese
2021-05-09 21:49:08 +08:00
quincylin1
e7847bd8b9
Visualise toy (#137)
* added img_norm_cfg key-value pair for toy data visualization

* add ori_filename
2021-04-28 12:42:08 +00:00
quincylin1
20cda15bcd
add img_norm_cfg to config (#136) 2021-04-28 17:56:00 +08:00
LvTianlei
7bf88d0e25
Add TPS (#117)
* Add TPS

* Update tps_preprocessor.py

* Add licence, change variable name and format

* renamed some parameters and add tests of ocr preprocessor

* renamed params

* Update tps_preprocessor.py

* add config file and readme for TPS
2021-04-28 07:39:13 +00:00
lizz
7dc5695d6d Add methods to readme
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 18:04:36 +08:00
Hongbin Sun
3993d77c95 fix iss21: error in crnn inference 2021-04-08 21:45:57 +08:00
lizz
0206781bb6
Fix readthedocs generation (#14)
* Fix readthedocs

* Fix doc

* md format
2021-04-08 15:49:46 +08:00
Hongbin Sun
c88126f0d2 fix #60: update readme 2021-04-07 10:19:36 +08:00
Hongbin Sun
0b2aa3aec6
fix #55: update nrtr readme (#56) 2021-04-06 14:39:06 +00:00
yuexy
4958f4df56
fix #39: update crnn & robustscanner. (#54) 2021-04-06 13:53:46 +00:00