Commit Graph

34 Commits (4c43f3fc4729df42d073e779f50488d63bbf7bd3)

Author SHA1 Message Date
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
yuexy b64f3906c0
add Dockerfile (#50) 2021-04-06 08:46:01 +00:00
Hongbin Sun 03720f46c3
fix #35: add nrtr readme; update nrtr config (#36) 2021-04-06 03:14:30 +00:00
Hongbin Sun 0f00378f9a
Feature/iss 33 (#34)
* fix #33: update dataset.md

* fix #33: pytest for transformer related
2021-04-05 23:54:57 +08:00
yuexy 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
Hongbin Sun cb1b2c2fa6
fix #24: rm img_meta from inference (#25) 2021-04-05 18:01:43 +08:00
lizz a48399aebe format again
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:39:31 +08:00
Hongbin Sun 4f5cde9120 fix #21: update readme of segocr 2021-04-05 16:34:09 +08:00
Hongbin Sun 3727cab712 fix #21: update sdmgr readme and config 2021-04-05 16:09:13 +08:00
lizz 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
Hongbin Sun 50287450ab
Merge pull request #5 from yuexy/feature/crnn_and_robustscanner
[feature]: add CRNN and RobustScanner
2021-04-03 00:55:27 +08:00
YueXy d594ab122e [feature]: add CRNN and RobustScanner 2021-04-03 00:52:49 +08:00
Hongbin Sun 4ecd0cea8a add sar, seg and other components 2021-04-02 23:54:57 +08:00