32 Commits

Author SHA1 Message Date
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
5a8859fe66
[Refactor] Refactor text detection config (#626)
* refactor textdet configs

* remove duplicate keys in _base_

* remove import from config

* syncbn to bn on cpu

* minimize change
2021-12-03 19:37:43 +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
Tong Gao
5654079839
[Fix] Fix DBNet_r50 config (#625) 2021-11-26 12:26:32 +00:00
Tong Gao
a9fde9146f
[Feature] Improve dbnet r50 by fixing std (#578) 2021-11-11 19:10:12 +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
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
7571763376
[Refactor] Use MMOCR's registry (#436)
* [Refactor] Use MMOCR's registry

1. Define MMOCR's registries as a child of MMDet's
2. Register all models to MMOCR's own registries
3. Modify some model configs so that some models in MMDet can be
   correctly located
4. Remove some outdated demo scripts

* add detectors
2021-08-19 19:17:15 +08:00
Tong Gao
4f7270e574
Fix #282: Support init_cfg & update depreciated configs (#365)
* update coco ref

* init_cfg for dbnet

* initcfg for mask_rcnn

* textsnake init_cfg

* fix dbnet

* panet initcfg

* psenet initcfg

* fcenet initcfg

* drrg initcfg

* add init_cfg to detectors

* update maskrcnn config file to support mmdet

* fix init_cfg of fce_head

* crnn initcfg

* init_weights in training

* nrtr initcfg

* robust_scanner initcfg

* sar init_cfg

* seg init_cfg

* tps_crnn init_cfg

* sdmgr initcfg

* ner init_cfg

* fix textsnake

* sdmgr initcfg

* move "pretrained" to "init_cfg" for config files

* Moduleslist update

* fix seg

* ner init_cfg

* fix base

* fix encode decode recognizer

* revert dbnet config

* fix crnn

* fix base.py

* fix robust_scanner

* fix panet

* fix test

* remove redundant init_weights() in fcehead

* clean up

* relex mmdet version in workflow

* Add dependency version check

* Update mmocr/models/textdet/dense_heads/pse_head.py

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-07-20 23:18:25 +08:00
Tong Gao
884755d05d
Fix #112: Remove the need of drop_orientation_info in data preprocessing steps (#375)
* ctw1500 ignore orientation

* restore maskrcnn config

* ignore_orientation support for icdar datasets

* update docs

* ignore orientation for total text

* Add LoadOCRImageFromFile

* Fix typo

* simplify design

* remove LoadOCRImageFromFile

* update chinese docs
2021-07-20 23:02:25 +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
Zyq-scut
d4ba5b8c9b
fix fcenet readme (#270) 2021-06-09 20:03:16 +08:00
Zyq-scut
18e7ecc379
add fce ic15 (#258) 2021-06-03 15:00:38 +08:00
Jianyong Chen
17aa9ecc7f
update metafile (#228) 2021-05-22 16:10:21 +08:00
Jianyong Chen
2414c65577
Add drrg (#189)
* merge drrg

* directory structure&fix redundant import

* docstrings

* fix isort

* drrg readme

* merge drrg

* directory structure&fix redundant import

* docstrings

* fix isort

* drrg readme

* add unittests&fix docstrings

* revert test_loss

* add unittest

* add unittests

* fix docstrings

* fix docstrings

* fix yapf

* fix yapf

* Update test_textdet_head.py

* Update test_textdet_head.py

* add unittests

* add unittests

* add unittests

* fix docstrings

* fix docstrings

* fix docstring

* fix unittests

* fix pytest

* fix pytest

* fix pytest

* fix variable names

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-18 11:15:47 +08:00
Zyq-scut
cbdd98a1e1
add fcenet (#133)
* add fcenet

* fix linting and code style

* fcenet finetune

* Update transforms.py

* Update fcenet_r50dcnv2_fpn_1500e_ctw1500.py

* Update fcenet_targets.py

* Update fce_loss.py

* fix

* add readme

* fix config

* Update fcenet_r50dcnv2_fpn_1500e_ctw1500.py

* fix

* fix readme

* fix readme

* Update test_loss.py

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-14 21:37:04 +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
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
a019058503
fix #51: revised readme.md for dbnet (#54)
* fix #51: revised readme.md for dbnet

* Update configs/textdet/dbnet/README.md

Co-authored-by: jeffreykuang <kuangzhh@gmail.com>
2021-04-13 11:53:57 +08:00
lizz
7dc5695d6d Add methods to readme
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 18:04:36 +08:00
Hongbin Sun
b1e20d8d39 fix #62: update panet readme 2021-04-07 12:38:37 +08:00
lizz
44ca9c2a61
Remove usage of \ (#49)
* Remove usage of \

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

* rebase

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

* typos

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

* Remove test dependency on tools/

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

* Remove usage of \

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

* rebase

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

* typos

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

* Remove test dependency on tools/

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

* typo

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

* KIE in keywords

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

* some renames

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

* kill isort skip

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

* aggregation discrimination

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

* aggregation discrimination

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

* tiny

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

* fix bug: model infer on cpu

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-04-06 12:16:46 +00:00
Hongbin Sun
03720f46c3
fix #35: add nrtr readme; update nrtr config (#36) 2021-04-06 03:14:30 +00: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
jeffreykuang
b10bb9eb6a
Enhance/synthtext pretrain (#20)
* fix 19: add synthtext pretrained model

* fix 19: setup.cfg linting
2021-04-04 21:15:21 +08:00
Hongbin Sun
12df13244d
fix #17: update ckpt path of psenet (#18) 2021-04-04 14:27:37 +08:00
Hongbin Sun
b7dac0f6fd
Merge pull request #7 from jeffreykuang/feature/textdet
[feature]: add textdet
2021-04-03 01:07:29 +08:00
jeffreykuang
31f1c007c0 [feature]: add textdet 2021-04-03 01:03:52 +08:00
Hongbin Sun
b8156a3a77 [feature]: add code for kie and textsnake config 2021-04-03 00:44:12 +08:00