195 Commits

Author SHA1 Message Date
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
Hongbin Sun
ecfc1c7ebf
fix #216: default ignore -100 (#221)
* fix #216: default ignore -100

* update ckpt
2021-05-20 21:53:49 +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
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
df5493a79e
refactor tps config (#135)
* refactor tps config

* recover tps

* add ckpt of tps
2021-05-12 06:14:24 +00:00
kang sheng
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
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
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
quincylin1
20cda15bcd
add img_norm_cfg to config (#136) 2021-04-28 17:56:00 +08:00
LvTianlei
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
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
3993d77c95 fix iss21: error in crnn inference 2021-04-08 21:45:57 +08:00
lizz
0206781bb6
Fix readthedocs generation (#14)
* Fix readthedocs

* Fix doc

* md format
2021-04-08 15:49:46 +08:00
Hongbin Sun
b1e20d8d39 fix #62: update panet readme 2021-04-07 12:38:37 +08:00
Hongbin Sun
c88126f0d2 fix #60: update readme 2021-04-07 10:19:36 +08:00
Hongbin Sun
0b2aa3aec6
fix #55: update nrtr readme (#56) 2021-04-06 14:39:06 +00:00
yuexy
4958f4df56
fix #39: update crnn & robustscanner. (#54) 2021-04-06 13:53:46 +00: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
yuexy
b64f3906c0
add Dockerfile (#50) 2021-04-06 08:46:01 +00:00
lizz
5ace2048ea out_size -> output_size
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-06 13:22:33 +08:00
Hongbin Sun
03720f46c3
fix #35: add nrtr readme; update nrtr config (#36) 2021-04-06 03:14:30 +00: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
yuexy
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
Hongbin Sun
cb1b2c2fa6
fix #24: rm img_meta from inference (#25) 2021-04-05 18:01:43 +08:00
lizz
a48399aebe format again
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:39:31 +08:00
lizz
49b52c9d8a f-str
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:37:14 +08:00
Hongbin Sun
4f5cde9120 fix #21: update readme of segocr 2021-04-05 16:34:09 +08:00
Hongbin Sun
3727cab712 fix #21: update sdmgr readme and config 2021-04-05 16:09:13 +08:00
Hongbin Sun
9af0bed144 fix #21: refactor kie dataset & add show_results 2021-04-05 16:07:29 +08: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
50287450ab
Merge pull request #5 from yuexy/feature/crnn_and_robustscanner
[feature]: add CRNN and RobustScanner
2021-04-03 00:55:27 +08:00
YueXy
d594ab122e [feature]: add CRNN and RobustScanner 2021-04-03 00:52:49 +08:00
Hongbin Sun
b8156a3a77 [feature]: add code for kie and textsnake config 2021-04-03 00:44:12 +08:00
Hongbin Sun
4ecd0cea8a add sar, seg and other components 2021-04-02 23:54:57 +08:00