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
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
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
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
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
b8f7ead74c
[Enhancement] Add copyright info ( #439 )
...
* add copyright info
2021-08-17 17:39:30 +08:00
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
f24be6c614
fix bug of dice loss: loss always 1 if empty target ( #408 )
2021-08-04 06:05:31 +00: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
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
97ac619e68
[Fix] Fix unit tests, remove temporary files ( #328 )
...
* fix unit tests, remove temp files
* use tempfile package and fix typo
* all files clear
2021-06-26 22:51:50 +08:00
d57f279083
[Fix] Correct isort setup ( #288 )
...
* isort
* remove tools from setup.cfg
2021-06-16 19:44:01 +08:00
0131b3290f
[Feature] Add onnx and tensorrt export tool ( #278 )
...
* add onnx and tensorrt export
* fix lint
* delete batch input to avoid dbnet error
* resolve unittest
* fix lint
* export unittest
2021-06-15 07:51:36 +00:00
18e7ecc379
add fce ic15 ( #258 )
2021-06-03 15:00:38 +08:00
0bd32889d4
fix #212 : improve drrg ( #224 )
2021-05-21 18:36:44 +08:00
8c60e4b4d1
update fcenet ( #214 )
...
* update fcenet
* fix typos
* fix
2021-05-18 21:52:34 +08:00
9e0595f70b
Test more on cpu ( #217 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 21:47:32 +08:00
673aadc355
Hbsun/feature iss205 ( #210 )
...
* fix #205 : remove act2fn
* fix pytest
2021-05-18 15:15:35 +08:00
24c590bb04
Ner task ( #148 )
...
* update ner standard code format
* add pytest
* fix pre-commit
* Annotate the dataset section
* fix pre-commit for dataset
* rm big files and add comments in dataset
* rename configs for ner task
* minor changes if metric
* Note modification
* fix pre-commit
* detail modification
* rm transform
* rm magic number
* fix warnings in pylint
* fix pre-commit
* correct help info
* rename model files
* rename err fixed
* 428_tag
* Adjust to more general pipline
* update unit test rate
* update
* Unit test coverage over 90% and add Readme
* modify details
* fix precommit
* update
* fix pre-commit
* update
* update
* update
* update result
* update readme
* update baseline config
* update config and small minor changes
* minor changes in readme and etc.
* back to original
* update toy config
* upload model and log
* fix pytest
* Modify the notes.
* fix readme
* Delete Chinese punctuation
* add demo and fix some logic and naming problems
* add To_tensor transformer for ner and load pretrained model in config
* delete extra lines
* split ner loss to MaskedCrossEntropyLoss and MaskedFocalLoss
* update config
* fix err
* updata
* modify noqa
* update new model report
* fix err in ner demo
* Update ner_dataset.py
* Update test_ner_dataset.py
* Update ner_dataset.py
* Update ner_transforms.py
* rm toy config and data
* add comment
* add empty
* fix conflict
* fix precommit
* fix pytest
* fix pytest err
* Update ner_dataset.py
* change dataset name to cluener2020
* move the postprocess in metric to convertor
* rm __init__ etc.
* precommit
* add discription in loss
* add auto download
* add http
* update
* remove some 'issert'
* replace unsqueeze
* update config
* update doc and bert.py
* update
* update demo code
Co-authored-by: weihuaqiang <weihuaqiang@sensetime.com>
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-18 11:33:51 +08:00
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
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
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
e29b5b8abe
Fix issue 122 ( #130 )
...
* fix #122 : textsnake targets adaptation
* fix #122 : textsnake targets adaptation
* add unittest
* fix format
* fix textsnake unittest on cpu
* fix unit test coverage
* add unit test
2021-04-29 11:02:57 +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
524498479f
add unit test for textsnake ( #61 )
2021-04-13 21:59:19 +08:00
344cc9a38d
Add unittest for kie ( #56 )
...
* add unittest for sdmgr
* fix #36 : update datasets.md
* unittest pass
2021-04-13 13:58:31 +08:00
c345255e49
remove test for textsnake ( #8 )
2021-04-08 03:07:22 +00:00
7acfabb482
fix pytest error ( #6 )
2021-04-08 01:48:15 +08:00
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
cbb4ec349b
Textsnake tests ( #51 )
...
* add textsnake unit tests
2021-04-06 18:20:43 +08:00
fbb5c8cda1
fix #40 : add unit test for recog config, transforms, etc. ( #44 )
2021-04-06 07:17:20 +00:00
c8793ac141
Add textdet unit tests ( #43 )
...
* Fix #41 : test fpn_cat
* Fix #41 : test fpn_cat
* Fix #41 : test fpn_cat
2021-04-06 07:16:27 +00:00
aa87b69f12
fix #37 : remove useless code ( #38 )
2021-04-06 03:40:48 +00:00
7e281521cf
skip cuda
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-06 10:58:45 +08:00
06fee8328b
fix #31 : pytest pass
2021-04-06 10:58:45 +08:00
b153cc01c8
Re-enable skipped test
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-06 10:57:25 +08: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
31f1c007c0
[feature]: add textdet
2021-04-03 01:03:52 +08:00
4ecd0cea8a
add sar, seg and other components
2021-04-02 23:54:57 +08:00
Tong Gao
Hongbin Sun
LvTianlei
Jiaqi Duan
AllentDan
Zyq-scut
Jianyong Chen
lizz
hq_wei
kang sheng
jeffreykuang
lizz