mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity
855 Commits 5 Branches 20 Tags
Commit Graph

130 Commits

Author SHA1 Message Date
liukuikun
7c3789d64e [PANLoss] pan loss 2022-07-21 10:51:02 +08:00
gaotongxiao
21d0dd71dc [TextSnake] Refactor textsnake loss 2022-07-21 10:51:02 +08:00
wangxinyu
bf7c738798 [TextSnake] TextSnake Neck 2022-07-21 10:51:02 +08:00
wangxinyu
acd2bcc452 [TextSnake] TextSnake Postprocessor 2022-07-21 10:51:02 +08:00
wangxinyu
f7731c43bd [TextSnake] TextSnake Head 2022-07-21 10:51:02 +08:00
liukuikun
a353a28a1a fix test path 2022-07-21 10:51:02 +08:00
jiangqing.vendor
a135580912 [MASTER] Add Master decoder 2022-07-21 10:51:01 +08:00
liukuikun
d5a2d20574 [Enchance] ce ignore char 2022-07-21 10:51:01 +08:00
wangxinyu
b44869059b [SATRN] SATRN Backbone 2022-07-21 10:51:01 +08:00
wangxinyu
401088913b [SATRN] SATRN Encoder 2022-07-21 10:51:01 +08:00
gaotongxiao
35e5138b5d Cleanup test tmp files in test crnn decoder 2022-07-21 10:51:01 +08:00
jiangqing.vendor
50f229d9fe [NRTR] NRTR Encoder 2022-07-21 10:51:01 +08:00
jiangqing.vendor
8614070e36 [NRTR] NRTR Decoder 2022-07-21 10:51:01 +08:00
jiangqing.vendor
d41921f03d [NRTR] NRTR backbone 2022-07-21 10:51:01 +08:00
gaotongxiao
b585dbcdd7 Fix score field in DBPostprocessor 2022-07-21 10:51:00 +08:00
liukuikun
38eef984c2 [Refactor]CELoss 2022-07-21 10:51:00 +08:00
gaotongxiao
8396b2014e Refactor FPNC 2022-07-21 10:51:00 +08:00
gaotongxiao
32ef9cc3cf [DBNet] Add DBHead 2022-07-21 10:51:00 +08:00
gaotongxiao
7a66a84b64 [DBNet] Add DBPostProcessor 2022-07-21 10:51:00 +08:00
gaotongxiao
cd3d173b18 [DBNet] Add DBLoss 2022-07-21 10:51:00 +08:00
gaotongxiao
747b2a14dc Refactor ResNet backbone 2022-07-21 10:51:00 +08:00
gaotongxiao
43c50eee82 fix loss 2022-07-21 10:51:00 +08:00
gaotongxiao
1e1da7b395 [DBNet] Add MaskedSmmothL1Loss, MasedBalancedBCELoss and MaskedDiceLoss 2022-07-21 10:50:59 +08:00
liukuikun
bbbefaeb31 build LabelData in baseprocessor 2022-07-21 10:50:59 +08:00
liukuikun
7582fdea41 [Refactor] CTCLoss 2022-07-21 10:50:59 +08:00
liukuikun
3aae157aec [Refactor] crnn decoder 2022-07-21 10:50:59 +08:00
liukuikun
4fd048aa24 [Fix] fix base recog loss 2022-07-21 10:50:59 +08:00
liukuikun
05e31e09bc [Fix] base decoder forget passing dictionary 2022-07-21 10:50:58 +08:00
gaotongxiao
be30df5d50 Rename TextRecog directories 2022-07-21 10:50:58 +08:00
gaotongxiao
f820a50752 Rename TextDet directories 2022-07-21 10:50:58 +08:00
liukuikun
a05e3f19c5 [Feature] TextRecogPostprocessor 2022-07-21 10:50:58 +08:00
liukuikun
0b5d2df310 [Refactor] BaseDecoder 2022-07-21 10:50:58 +08:00
liukuikun
6cd38a038f [Refactor] base recog loss 2022-07-21 10:50:57 +08:00
gaotongxiao
cb85f857aa Add BaseTextDetPostProcessor 2022-07-21 10:50:57 +08:00
liukuikun
c920edfb3a [Refactor] split labelconverter to Dictionary 2022-07-21 10:50:56 +08:00
gaotongxiao
69e6c80558 move tests to old_tests, add empty test folders 2022-07-21 10:50:55 +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
Xinyu Wang
fbc138d2d3
[Feature] Add DBNet++ (#973) 
* add dbnet++

* fix docstring

* fix some param names

* fix

* fix docstring

* add init

* add doc; remove configs

* add dbnet++ to readme

* fix readme

* update config

* update readme

* update readme

* update ocr.py

* update metafile.yml

* update readme

* update readme

* move to dbnetpp

* fix paths

* fix head level

* fix typo

* update demo.md

* Update configs/textdet/dbnetpp/README.md

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

* fix typo

* fix link
 2022-05-05 14:39:20 +08:00
Tong Gao
37833ada4b
[Fix] Support loading space character from dict file (#854) 
* [Feature] Support loading dict file with space character

* fix tests

* clean up \\r in tests

* add DICT37 and DICT91

* update docstr
 2022-03-22 17:44:32 +08:00
JiangQing
72f9454573
[Refactor] Refactor ResNets (#809) 2022-03-18 10:36:03 +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
7874531db3
[Refactor] refactor textdet postprocess (#640) 
* refactor textdet postprocess using build_postprocessor

* update docstring

* update docstring

* update postprocess docstring

* compatible with previous configs

* rename to BaseTextDetector

* rename to DRRGPostprocessor

* fix

* update warning

* fix

* fix conflict

* rename back

* rename warpper to utils

* fix doc
 2021-12-14 17:35:44 +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
Hongbin Sun
9f42d78db7
[Feature] Save filename in text detection test results (#570) 
* save filename in textdet test results

* fix pytest
 2021-11-11 14:07:07 +08: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
Tong Gao
b8f7ead74c
[Enhancement] Add copyright info (#439) 
* add copyright info
 2021-08-17 17:39:30 +08:00
Tong Gao
7bbb14f0d1
[Fix] Replace SyncBN with BN for inference (#420) 
* add revert_sync_batchnorm

* replace SyncBN in inference and test scripts

* add tests

* hide BatchNormXd
 2021-08-10 22:19:17 +08:00
Jiaqi Duan
f24be6c614
fix bug of dice loss: loss always 1 if empty target (#408) 2021-08-04 06:05:31 +00: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
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