mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
273 Commits
5 Branches
20 Tags
54 MiB
Commit Graph

23 Commits (c1ae3a41e5fafc0e64bece2be108f1867214cd26)

Author SHA1 Message Date
Hongbin Sun 44fed8590b
fix bug: remove tail whitespace from gt label (#358) 
* fix bug: remove tail whitespace from gt label

* remove useless

* refactor
 2021-07-09 13:24:34 +00: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
lizz 8999c56159
Fix kie_test_imgs script for novisual (#319) 
* Fix kie_test_imgs script for novisual

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

* Fix test

Signed-off-by: lizz <lizz@sensetime.com>
 2021-06-24 07:22:34 +00:00
Hongbin Sun 7c4eb7e056
fix #253: speedup; fix assert (#263) 
* fix #253: speedup; fix assert

* fix #253: speedup; fix assert

* add unittest for bug fix block
 2021-06-23 08:16:57 +00:00
AllentDan d57f279083
[Fix] Correct isort setup (#288) 
* isort

* remove tools from setup.cfg
 2021-06-16 19:44:01 +08:00
lizz 36e92ebe70
Speed up sort_vertex (#239) 
* Speed up sort_vertex

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

* Fix test

Signed-off-by: lizz <lizz@sensetime.com>
 2021-05-25 19:58:05 +08:00
Zyq-scut 8c60e4b4d1
update fcenet (#214) 
* update fcenet

* fix typos

* fix
 2021-05-18 21:52:34 +08:00
hq_wei 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
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
jeffreykuang 47896a3f80
Unit tests of dbnet transform (#163) 
* Add unit test for dbnet transforms

* Add unit test for dbnet transforms

* Add unit test for dbnet transforms
 2021-05-11 15:21:14 +08:00
lizz 5054a3f78d
Implement iter protocol for Loader (#150) 
* Implement iter protocal for Loader

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

* Test iter protocal

Signed-off-by: lizz <lizz@sensetime.com>
 2021-05-06 09:51:29 +00:00
lizz 47965516f0
Improve KIE (#153) 
* Improve KIE
* allow data do not contain label
* Make relation float32

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

* Add test

Signed-off-by: lizz <lizz@sensetime.com>
 2021-05-06 09:51:10 +00:00
Jianyong Chen 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
Hongbin Sun 43dcb32d4f
Hbsun/end2end demo (#105) 
* add end2end demo

* fix typo

* pad box

* fix bug of crnn

* fix polygon

* update docstring

* fix bug of polygon

* updare demo api

* fix except

* rename

* fix with comments
 2021-04-22 20:42:42 +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
Theo Chan cbb4ec349b
Textsnake tests (#51) 
* add textsnake unit tests
 2021-04-06 18:20:43 +08:00
Hongbin Sun fbb5c8cda1
fix #40: add unit test for recog config, transforms, etc. (#44) 2021-04-06 07:17:20 +00:00
lizz 09ffd284ee Remove test dependency on tools 
Signed-off-by: lizz <lizz@sensetime.com>
 2021-04-06 10:57:25 +08:00
lizz 2ad699e099 fix 
Signed-off-by: lizz <lizz@sensetime.com>
 2021-04-06 10:57:25 +08: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
jeffreykuang 31f1c007c0 [feature]: add textdet 2021-04-03 01:03:52 +08:00
Hongbin Sun 4ecd0cea8a add sar, seg and other components 2021-04-02 23:54:57 +08:00