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
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
6d97d8286e
Update nrtr batch size ( #370 )
2021-07-16 15:06:36 +08:00
1a5a880abb
fix nrtr cfg ( #356 )
2021-07-08 13:30:58 +08:00
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
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
ae90deab68
fix bug in sdmgr config ( #316 )
2021-06-23 16:58:56 +08:00
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
50a25e718a
update ckpt link ( #287 )
2021-06-15 20:28:09 +08:00
8dcc56955f
Ignore img processing for KIE without visual ( #260 )
...
* Ignore img processing for KIE without visual
Signed-off-by: lizz <lizz@sensetime.com>
* Add ResizeNoImg
Signed-off-by: lizz <lizz@sensetime.com>
* Tune speed
Signed-off-by: lizz <lizz@sensetime.com>
2021-06-10 10:07:33 +08:00
d4ba5b8c9b
fix fcenet readme ( #270 )
2021-06-09 20:03:16 +08:00
18e7ecc379
add fce ic15 ( #258 )
2021-06-03 15:00:38 +08:00
17aa9ecc7f
update metafile ( #228 )
2021-05-22 16:10:21 +08:00
ecfc1c7ebf
fix #216 : default ignore -100 ( #221 )
...
* fix #216 : default ignore -100
* update ckpt
2021-05-20 21:53:49 +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
1240169f18
update metafile ( #183 )
2021-05-14 00:17:33 +08:00
7e0d48df11
fix metafiles ( #182 )
2021-05-13 17:28:59 +08:00
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
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
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
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
20cda15bcd
add img_norm_cfg to config ( #136 )
2021-04-28 17:56:00 +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
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
7dc5695d6d
Add methods to readme
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 18:04:36 +08:00
3993d77c95
fix iss21: error in crnn inference
2021-04-08 21:45:57 +08:00
0206781bb6
Fix readthedocs generation ( #14 )
...
* Fix readthedocs
* Fix doc
* md format
2021-04-08 15:49:46 +08:00
b1e20d8d39
fix #62 : update panet readme
2021-04-07 12:38:37 +08:00
c88126f0d2
fix #60 : update readme
2021-04-07 10:19:36 +08:00
0b2aa3aec6
fix #55 : update nrtr readme ( #56 )
2021-04-06 14:39:06 +00:00
4958f4df56
fix #39 : update crnn & robustscanner. ( #54 )
2021-04-06 13:53:46 +00: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
b64f3906c0
add Dockerfile ( #50 )
2021-04-06 08:46:01 +00:00
5ace2048ea
out_size -> output_size
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-06 13:22:33 +08:00
03720f46c3
fix #35 : add nrtr readme; update nrtr config ( #36 )
2021-04-06 03:14:30 +00: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
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
cb1b2c2fa6
fix #24 : rm img_meta from inference ( #25 )
2021-04-05 18:01:43 +08:00
a48399aebe
format again
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:39:31 +08:00
49b52c9d8a
f-str
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 16:37:14 +08:00
4f5cde9120
fix #21 : update readme of segocr
2021-04-05 16:34:09 +08:00
3727cab712
fix #21 : update sdmgr readme and config
2021-04-05 16:09:13 +08:00
9af0bed144
fix #21 : refactor kie dataset & add show_results
2021-04-05 16:07:29 +08:00
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
b10bb9eb6a
Enhance/synthtext pretrain ( #20 )
...
* fix 19: add synthtext pretrained model
* fix 19: setup.cfg linting
2021-04-04 21:15:21 +08:00
12df13244d
fix #17 : update ckpt path of psenet ( #18 )
2021-04-04 14:27:37 +08:00
b7dac0f6fd
Merge pull request #7 from jeffreykuang/feature/textdet
...
[feature]: add textdet
2021-04-03 01:07:29 +08:00
31f1c007c0
[feature]: add textdet
2021-04-03 01:03:52 +08:00
50287450ab
Merge pull request #5 from yuexy/feature/crnn_and_robustscanner
...
[feature]: add CRNN and RobustScanner
2021-04-03 00:55:27 +08:00
d594ab122e
[feature]: add CRNN and RobustScanner
2021-04-03 00:52:49 +08:00
b8156a3a77
[feature]: add code for kie and textsnake config
2021-04-03 00:44:12 +08:00
4ecd0cea8a
add sar, seg and other components
2021-04-02 23:54:57 +08:00
Hongbin Sun
Tong Gao
lizz
Zyq-scut
Jianyong Chen
hq_wei
quincylin1
kang sheng
LvTianlei
lizz
yuexy
jeffreykuang
YueXy