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
Hongbin Sun
Tong Gao
Shouping Shan
LvTianlei
quincylin1
kang sheng
lizz
lizz
yuexy
YueXy