dc84187311
[TODO] Replace loss_module with module_loss
2022-07-21 10:57:31 +08:00
3709c7b03a
[Metafiles] Update Metafiles for FCENet, MaskRCNN, NRTR, MASTER
2022-07-21 10:57:31 +08:00
914c8af7bf
Revert "[TODO] Add LoadImageFromLMDB"
...
This reverts commit e716ae726f007f79effdf2d45b4955a887f3c1e3
2022-07-21 10:57:30 +08:00
19958fbf6f
[TODO] Add LoadImageFromLMDB
2022-07-21 10:57:17 +08:00
2d478ea244
Add ABINet cfg
2022-07-21 10:55:46 +08:00
77ffe8fb00
[Config] scanner config
2022-07-21 10:55:46 +08:00
ca35c78e69
[Config] SAR config
2022-07-21 10:51:03 +08:00
b406f3785f
[MASTER] Add MASTER config
2022-07-21 10:51:02 +08:00
3992f0d78e
[SATRN] SATRN Config
2022-07-21 10:51:01 +08:00
4f2ec6de71
[NRTR] NRTR config
2022-07-21 10:51:01 +08:00
f1eebe9e34
[CRNN] CRNN config
2022-07-21 10:51:00 +08:00
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
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
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
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
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
5176e1bd5a
[Fix] fix link ( #726 )
2022-01-12 10:25:07 +08:00
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
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
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
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
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
5fb1268a06
unify task names in metafiles ( #548 )
2021-11-03 15:03:37 +08:00
dab5725d70
fix paper field in metafiles ( #550 )
2021-11-03 15:02:31 +08:00
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
abfe75dc62
[Fix] Update checkpoint links for SATRN ( #518 )
2021-10-10 11:26:37 +08:00
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
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
3707d67106
support batch inference for crnn and segocr ( #407 )
...
* support batch inference for crnn and segocr
2021-08-03 16:52:58 +08:00
6d97d8286e
Update nrtr batch size ( #370 )
2021-07-16 15:06:36 +08:00
1a5a880abb
fix nrtr cfg ( #356 )
2021-07-08 13:30:58 +08:00
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
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
1240169f18
update metafile ( #183 )
2021-05-14 00:17:33 +08:00
7e0d48df11
fix metafiles ( #182 )
2021-05-13 17:28:59 +08:00
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
df5493a79e
refactor tps config ( #135 )
...
* refactor tps config
* recover tps
* add ckpt of tps
2021-05-12 06:14:24 +00:00
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
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
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
20cda15bcd
add img_norm_cfg to config ( #136 )
2021-04-28 17:56:00 +08:00
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
7dc5695d6d
Add methods to readme
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 18:04:36 +08:00
3993d77c95
fix iss21: error in crnn inference
2021-04-08 21:45:57 +08:00
0206781bb6
Fix readthedocs generation ( #14 )
...
* Fix readthedocs
* Fix doc
* md format
2021-04-08 15:49:46 +08:00
c88126f0d2
fix #60 : update readme
2021-04-07 10:19:36 +08:00
0b2aa3aec6
fix #55 : update nrtr readme ( #56 )
2021-04-06 14:39:06 +00:00
4958f4df56
fix #39 : update crnn & robustscanner. ( #54 )
2021-04-06 13:53:46 +00:00
b64f3906c0
add Dockerfile ( #50 )
2021-04-06 08:46:01 +00:00
03720f46c3
fix #35 : add nrtr readme; update nrtr config ( #36 )
2021-04-06 03:14:30 +00:00
0f00378f9a
Feature/iss 33 ( #34 )
...
* fix #33 : update dataset.md
* fix #33 : pytest for transformer related
2021-04-05 23:54:57 +08:00
3b443cf6f9
[feature]: add nrtr ( #28 )
...
* [feature]: add nrtr
* Rename nrtr_top_dataset.py to nrtr_toy_dataset.py
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-04-05 20:19:45 +08:00
cb1b2c2fa6
fix #24 : rm img_meta from inference ( #25 )
2021-04-05 18:01:43 +08:00
a48399aebe
format again
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:39:31 +08:00
4f5cde9120
fix #21 : update readme of segocr
2021-04-05 16:34:09 +08:00
3727cab712
fix #21 : update sdmgr readme and config
2021-04-05 16:09:13 +08:00
9d62bdf84c
Format readme ( #23 )
...
* Format readme
Signed-off-by: lizz <lizz@sensetime.com>
* try
Signed-off-by: lizz <lizz@sensetime.com>
* Remove redudant config link
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:06:06 +08:00
50287450ab
Merge pull request #5 from yuexy/feature/crnn_and_robustscanner
...
[feature]: add CRNN and RobustScanner
2021-04-03 00:55:27 +08:00
d594ab122e
[feature]: add CRNN and RobustScanner
2021-04-03 00:52:49 +08:00
4ecd0cea8a
add sar, seg and other components
2021-04-02 23:54:57 +08:00
jiangqing.vendor
gaotongxiao
liukuikun
wangxinyu
liukuikun
Jianquan Ye
Hongbin Sun
Shouping Shan
LvTianlei
quincylin1
kang sheng
lizz
lizz
yuexy
YueXy