Commit Graph

57 Commits (a86ab115e4fae0066e5a6a4167c27c3e233e7470)

Author SHA1 Message Date
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
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
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
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
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
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
Hongbin Sun a7a64f316b
update install (#222) 2021-05-20 12:51:22 +08:00
Hongbin Sun 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
Hongbin Sun 673aadc355
Hbsun/feature iss205 (#210)
* fix #205: remove act2fn

* fix pytest
2021-05-18 15:15:35 +08:00
Hongbin Sun 11768d02d4
update dataset doc (#204) 2021-05-18 11:43:29 +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
jeffreykuang 94cd52dc4a
simplify installation procedure (#188)
* postproc

* mv postproc of textdet in mmcv

* linting

* change deps mmcv 1.3.4

* update deps

* linting

* Update configs/textdet/panet/panet_r18_fpem_ffm_600e_icdar2015.py

* Update docs/changelog.md

* Update mmocr/models/textdet/postprocess/__init__.py

* Update mmocr/version.py

* resize&recompress jpg

* Update changelog.md

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-15 10:34:37 +08:00
Hongbin Sun e2643df439
move demo image to demo/resources (#171)
* move demo image to demo/resources

* update index.rst

* revert

* update github link for demo image
2021-05-12 20:35:39 +08:00
Hongbin Sun 3b9306ca79
Hbsun/demo doc (#168)
* move demo.md to demo/docs

* update link in md
2021-05-12 09:57:53 +08:00
Hongbin Sun 45d22b12c6
update demo docs (#166)
* update demo docs

* update docs
2021-05-11 21:13:19 +08:00
quincylin1 3ccec09106
add mmocr demo documentation (#151)
* add mmocr demo documentation

* Update text_det_demo.md

* Update text_recog_demo.md

* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-11 16:38: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
Hongbin Sun 360082635b
fix #93: update ic17 download url (#94) 2021-04-19 13:29:36 +08:00
Hongbin Sun 096c7c11b3
update mmdet install (#84)
* update mmdet install

* update install.md

* fix version of mmcv and mmdet
2021-04-17 22:48:15 +08:00
Hongbin Sun 1a129a1e98
add svt converter (#65)
* add svt converter

* fix str fmt

* fix str fmt

* update convert script
2021-04-14 18:33:14 +08:00
Hongbin Sun 344cc9a38d
Add unittest for kie (#56)
* add unittest for sdmgr

* fix #36: update datasets.md

* unittest pass
2021-04-13 13:58:31 +08:00
Hongbin Sun 1176b485fa fix name mistake 2021-04-10 17:10:50 +08:00
lizz 87ccebcd91
Update datasets.md 2021-04-10 00:28:12 +08:00
lizz 8097113e17
Update datasets.md 2021-04-10 00:27:17 +08:00
Hongbin Sun bb51a41bcd rename to datasets 2021-04-09 23:50:33 +08:00
Hongbin Sun fca6986f5c remove duplicate col 2021-04-09 22:53:21 +08:00
Hongbin Sun 6fe6dac367 fix lint 2021-04-09 22:21:38 +08:00
Hongbin Sun 93483615b7 fix api vis bug: sphinx version 2021-04-09 22:18:26 +08:00
Hongbin Sun b3aaa27c6f rename datasets.md 2021-04-09 21:08:19 +08:00
lizz 912a2cc9d1
Mock more (#37)
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-09 16:51:25 +08:00
lizz 3282e433f9
Fix readthedocs (#35)
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-09 16:39:21 +08:00
Hongbin Sun de564ef7cb update changelog 2021-04-09 14:38:34 +08:00
Hongbin Sun 56ed699645 update install.md 2021-04-09 10:25:40 +08:00
Hongbin Sun 0cfb59867c update install.md 2021-04-09 00:10:55 +08:00
Hongbin Sun cb6a0d5c99 update docs 2021-04-08 18:05:46 +08:00
lizz 0206781bb6
Fix readthedocs generation (#14)
* Fix readthedocs

* Fix doc

* md format
2021-04-08 15:49:46 +08:00
Hongbin Sun b17304edeb
update readme (#9)
Co-authored-by: sunhongbin <sunhongbin@sensetime.com>
2021-04-08 03:52:43 +00:00
Hongbin Sun 923ee16bf7
fix typo (#7) 2021-04-08 02:23:03 +00:00
Hongbin Sun 47f5906f0a
remove markdown toc (#3)
* remove toc from md

* remove toc
2021-04-08 01:12:01 +08:00
Hongbin Sun 8622fe8a3d
fix #1: update steps of contributing (#2)
* fix #1: update steps of contributing

* fix #1: update docs

* fix #1: contributing
2021-04-08 00:54:30 +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
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
lizz 7471c8cecd
Fix typos (#26)
* Fix typos

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

* Ohh

Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 20:16:13 +08:00
Hongbin Sun 3727cab712 fix #21: update sdmgr readme and config 2021-04-05 16:09:13 +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
Hongbin Sun d48f212baa
fix #13: rename section title (#16) 2021-04-04 14:20:31 +08:00
Hongbin Sun ef59ab10f5 fix #13: rename section title 2021-04-04 13:38:40 +08:00