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
e6cb750922
add TextOCR dataset converter ( #293 )
...
* textocr converter for text recog
* textocr converter for text detection
* update documentation
* remove unnecessary garbage collection lines
* multi-processing textocr converter
* json->mmcv, fix documentation
2021-06-21 03:06:10 +00:00
9a2a8aa662
fix import in demo ( #304 )
2021-06-21 03:04:26 +00:00
9582f93c70
Support ocr box stitching ( #290 )
...
* ocr box stitching
* rename a varible
* add a demo
* move functions to box_util
add test
2021-06-19 19:28:14 +08:00
a86ab115e4
fix unit tests ( #300 )
2021-06-18 18:27:20 +08:00
f1b003ddb1
[Feature] Add deployment evaluation ( #291 )
...
* add deployment evaluation
* fix lint
* remove cpu unit tests for trt and onnx
* use pytest.mark to skip cpu unit test
* move to mmocr/core
* emm... renamed to wrappers
* renamed to deploy_utils
* renamed unit test to test_deploy_utils
* fix lint
* using pytest.mark.importorskip
2021-06-18 12:27:10 +08:00
d57f279083
[Fix] Correct isort setup ( #288 )
...
* isort
* remove tools from setup.cfg
2021-06-16 19:44:01 +08:00
50a25e718a
update ckpt link ( #287 )
2021-06-15 20:28:09 +08:00
87a7dcee0a
fix #279 : save detect results ( #281 )
...
* fix #279 : save detect results
* rename
* set device as arg
* rm bash file
2021-06-15 18:30:09 +08:00
3bfbb2b619
fix #113 : textsnake decode error ( #264 )
...
* fix #113 : textsnake decode error
* fix unittest test cases
2021-06-15 08:09:05 +00: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
07a21fd716
fix #284 : str only for separator ( #285 )
2021-06-15 10:50:48 +08:00
01d8d63be9
Update readthedocs.txt ( #276 )
2021-06-12 15:28:00 +08:00
d7fa9544e6
added totaltext recog converter ( #273 )
...
* added totaltext recog converter
* modified datasets.md and totaltext_converter.py
* added Note to datasets.md
* deleted comments
2021-06-11 11:09:35 +08:00
9fe1ad5410
dbnet dynamic ( #269 )
2021-06-10 13:06:03 +00: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
bb4447585f
Add benchmark_processing script ( #261 )
...
* Flush data script
Signed-off-by: lizz <lizz@sensetime.com>
* Update tools/run_data.py
Co-authored-by: jeffreykuang <kuangzhanghui@sensetime.com>
* Update tools/run_data.py
Co-authored-by: jeffreykuang <kuangzhanghui@sensetime.com>
* Update tools/run_data.py
Co-authored-by: jeffreykuang <kuangzhanghui@sensetime.com>
* rename to benchmark_processing
Signed-off-by: lizz <lizz@sensetime.com>
Co-authored-by: jeffreykuang <kuangzhanghui@sensetime.com>
2021-06-09 17:38:53 +08:00
179ab3719c
Remove pre-generated annotation files for ctw1500 ( #268 )
...
* Remove pre-generated annotation files for ctw1500
* fix disappeared ICDAR2015 in datasets.md
2021-06-09 07:00:58 +00:00
271129f812
Feature/iss 262 ( #266 )
...
* fix issue#262
* fix #262 : modified totaltext_converter and added totaltext for datasets.md
* fix issue#262: modified datasets.md
* fix issue#262: removed download json
* Update totaltext_converter.py
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-06-08 13:13:22 +00:00
71208cb8e7
fixed dead links of FCENet Readme ( #267 )
2021-06-08 17:52:32 +08:00
b1741dee57
Fix readme crnn link ( #265 )
...
* fix readme crnn link
* fix readme crnn-CN link
2021-06-08 06:09:47 +00:00
18e7ecc379
add fce ic15 ( #258 )
2021-06-03 15:00:38 +08:00
4882c8a317
dataset preparation docs ( #255 )
2021-06-01 21:59:40 +08:00
70715cee42
fix readthedoc ( #247 )
2021-05-27 21:58:11 +08:00
ef4402621f
add backend for resizeocr ( #244 )
...
* add backend for resizeocr
* update
2021-05-27 20:31:10 +08:00
c9d151e757
Fix installing requirements with pip and mim ( #242 )
...
* add requirements
* add requirements
2021-05-27 15:59:13 +08:00
1aa86fbda9
Support distributed evaluation of KIE ( #234 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-27 10:31:11 +08:00
b863bbca5e
use mmcv instead ( #237 )
...
* use mmcv instead
* update
2021-05-25 19:58:32 +08:00
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
1e527e77be
Make dirs for list_to_file ( #235 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-25 10:23:24 +08:00
b10b6408ef
Add list_from_file and list_to_file ( #226 )
...
* Add list_from_file and list_to_file
Signed-off-by: lizz <lizz@sensetime.com>
* Add test list_to_file and list_from_file
* more
* Fix tests
2021-05-24 06:01:42 +00:00
17aa9ecc7f
update metafile ( #228 )
2021-05-22 16:10:21 +08:00
5121b9fd94
Support cpu for ocr demo ( #227 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-22 15:51:21 +08:00
0bd32889d4
fix #212 : improve drrg ( #224 )
2021-05-21 18:36:44 +08:00
ecfc1c7ebf
fix #216 : default ignore -100 ( #221 )
...
* fix #216 : default ignore -100
* update ckpt
2021-05-20 21:53:49 +08:00
a7a64f316b
update install ( #222 )
2021-05-20 12:51:22 +08:00
b413292738
update link ( #220 )
2021-05-19 10:27:49 +08:00
af3cb8d5d7
update publish yml ( #218 )
2021-05-18 22:58:21 +08:00
62c452359f
update changelog for v0.2.0 ( #208 )
...
* update changelog for v0.2.0
* update changelog
* update
2021-05-18 21:57:27 +08:00
8c60e4b4d1
update fcenet ( #214 )
...
* update fcenet
* fix typos
* fix
2021-05-18 21:52:34 +08:00
e0e0e648fd
Support cpu in kie visulization ( #215 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 21:48:16 +08:00
9e0595f70b
Test more on cpu ( #217 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 21:47:32 +08:00
c083334f73
update readme ( #211 )
...
* update readme
* updare README_zh-CN.md
* update link
2021-05-18 16:25:23 +08:00
491866ea27
Make tools executable ( #209 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 15:19:08 +08:00
673aadc355
Hbsun/feature iss205 ( #210 )
...
* fix #205 : remove act2fn
* fix pytest
2021-05-18 15:15:35 +08:00
06b75780a0
Fix typos ( #207 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 05:44:52 +00:00
11768d02d4
update dataset doc ( #204 )
2021-05-18 11:43:29 +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
Hongbin Sun
GT
AllentDan
Jianyong Chen
lizz
quincylin1
Zyq-scut
RangiLyu
hq_wei