Commit Graph

273 Commits (c1ae3a41e5fafc0e64bece2be108f1867214cd26)
 

Author SHA1 Message Date
Tong Gao c1ae3a41e5
Bump version to 0.2.1 (#378) 2021-07-20 23:19:07 +08:00
Tong Gao d8cb48ab46
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331)
* update pytorch req and dockerfile

* Update dependency requirement

* update readme for 0.2.1

* update change log

* update release date
2021-07-20 23:18:47 +08:00
Tong Gao 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
Tong Gao 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
Hongbin Sun 6d97d8286e
Update nrtr batch size (#370) 2021-07-16 15:06:36 +08:00
Tong Gao 0effc39c5f
[Docs] fix dead links, remove invalid TOC in CN docs (#367) 2021-07-15 08:28:36 +00:00
SamAyala 87bd295a20
[Feature] Add User friendly API (#366) 2021-07-15 14:18:11 +08:00
Zichen Tian 782e966c26
fix #347: add docs_zh_CN/datasets.md (#362)
* fix #347: add docs_zh_CN/datasets.md

* fix #347: translation correction
2021-07-13 12:54:47 +08:00
Tong Gao 02e3b98684
fix syntext_converter (#361) 2021-07-12 02:07:50 +00:00
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
quincylin1 243f47dc03
add totaltext for recog and det (#357)
* add totaltext for recog and det

* add setup

* fix doc

* fix based on comments
2021-07-08 21:52:50 +08:00
Tong Gao 1a5a880abb
fix nrtr cfg (#356) 2021-07-08 13:30:58 +08:00
Tong Gao 500b687469
[Docs] Fix missed deployment in en docs and add language switcher to the sidebar (#355)
* fix deployment and add language switcher

* change way to switch language and add chinese toc

* update zh_CN datasets.md (to be translated)
2021-07-07 16:21:52 +08:00
Tong Gao 68df4fbe80
[Feature] Add synthtext converter and update docs (#351)
* Add synthtext converter and update docs

* minor docs fix
2021-07-07 15:54:29 +08:00
Tong Gao 4fcff1f613
[Docs] Add cn docs framework (#353)
* add CN demo docs

* add deployment.md to docs

* add placeholder CN docs

* Add language switching hint
2021-07-07 14:13:27 +08:00
Antonio Lanza 19aefa1ae1
Added support for ConcatDataset with two workflows (#348) 2021-07-06 16:37:37 +08: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
Tong Gao 4559854af8
Fix metafiles (#342) 2021-07-01 10:59:57 +08:00
Zyq-scut a7a6c17bb5
fcenet fix bug (#334)
* fix bug

* to_device
2021-06-30 08:41:55 +08:00
miwojc 57e752e9b2
Update MMOCR_Tutorial.ipynb (#330)
add change directory to `mmocr` after git clone: `%cd mmocr`
"!pip install -r requirements.txt\n", was failing in colab as after git clone one needs to change dir to mmocr.
2021-06-29 17:00:47 +08:00
AllentDan 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
Tong Gao 079a93de4e
update mmcv installation method (#323) 2021-06-25 10:38:20 +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 ae90deab68
fix bug in sdmgr config (#316) 2021-06-23 16:58:56 +08: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
Hongbin Sun 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
GT 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
GT 9a2a8aa662
fix import in demo (#304) 2021-06-21 03:04:26 +00:00
GT 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
AllentDan a86ab115e4
fix unit tests (#300) 2021-06-18 18:27:20 +08:00
AllentDan 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
AllentDan d57f279083
[Fix] Correct isort setup (#288)
* isort

* remove tools from setup.cfg
2021-06-16 19:44:01 +08:00
Hongbin Sun 50a25e718a
update ckpt link (#287) 2021-06-15 20:28:09 +08:00
Hongbin Sun 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
Jianyong Chen 3bfbb2b619
fix #113: textsnake decode error (#264)
* fix #113: textsnake decode error

* fix unittest test cases
2021-06-15 08:09:05 +00:00
AllentDan 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
Hongbin Sun 07a21fd716
fix #284: str only for separator (#285) 2021-06-15 10:50:48 +08:00
lizz 01d8d63be9
Update readthedocs.txt (#276) 2021-06-12 15:28:00 +08:00
quincylin1 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
AllentDan 9fe1ad5410
dbnet dynamic (#269) 2021-06-10 13:06:03 +00:00
lizz 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
Zyq-scut d4ba5b8c9b
fix fcenet readme (#270) 2021-06-09 20:03:16 +08:00
lizz 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
GT 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
quincylin1 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
GT 71208cb8e7
fixed dead links of FCENet Readme (#267) 2021-06-08 17:52:32 +08:00
AllentDan b1741dee57
Fix readme crnn link (#265)
* fix readme crnn link

* fix readme crnn-CN link
2021-06-08 06:09:47 +00:00
Zyq-scut 18e7ecc379
add fce ic15 (#258) 2021-06-03 15:00:38 +08:00
Hongbin Sun 4882c8a317
dataset preparation docs (#255) 2021-06-01 21:59:40 +08:00
Hongbin Sun 70715cee42
fix readthedoc (#247) 2021-05-27 21:58:11 +08:00