Commit Graph

223 Commits (ef4402621fea34da4c2f28cd476b9e6e795b13a7)
 

Author SHA1 Message Date
Hongbin Sun ef4402621f
add backend for resizeocr (#244)
* add backend for resizeocr

* update
2021-05-27 20:31:10 +08:00
RangiLyu c9d151e757
Fix installing requirements with pip and mim (#242)
* add requirements

* add requirements
2021-05-27 15:59:13 +08:00
lizz 1aa86fbda9
Support distributed evaluation of KIE (#234)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-27 10:31:11 +08:00
Hongbin Sun b863bbca5e
use mmcv instead (#237)
* use mmcv instead

* update
2021-05-25 19:58:32 +08:00
lizz 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
lizz 1e527e77be
Make dirs for list_to_file (#235)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-25 10:23:24 +08:00
lizz 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
Jianyong Chen 17aa9ecc7f
update metafile (#228) 2021-05-22 16:10:21 +08:00
lizz 5121b9fd94
Support cpu for ocr demo (#227)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-22 15:51:21 +08:00
Jianyong Chen 0bd32889d4
fix #212: improve drrg (#224) 2021-05-21 18:36:44 +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
Hongbin Sun a7a64f316b
update install (#222) 2021-05-20 12:51:22 +08:00
Hongbin Sun b413292738
update link (#220) 2021-05-19 10:27:49 +08:00
Hongbin Sun af3cb8d5d7
update publish yml (#218) 2021-05-18 22:58:21 +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
Zyq-scut 8c60e4b4d1
update fcenet (#214)
* update fcenet

* fix typos

* fix
2021-05-18 21:52:34 +08:00
lizz e0e0e648fd
Support cpu in kie visulization (#215)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 21:48:16 +08:00
lizz 9e0595f70b
Test more on cpu (#217)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 21:47:32 +08:00
Hongbin Sun c083334f73
update readme (#211)
* update readme

* updare README_zh-CN.md

* update link
2021-05-18 16:25:23 +08:00
lizz 491866ea27
Make tools executable (#209)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 15:19:08 +08:00
Hongbin Sun 673aadc355
Hbsun/feature iss205 (#210)
* fix #205: remove act2fn

* fix pytest
2021-05-18 15:15:35 +08:00
lizz 06b75780a0
Fix typos (#207)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 05:44:52 +00: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
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
quincylin1 ed6b3b890a
modified mmocr colab (#199)
* modified mmocr colab

* modified mmocr colab

* modified mmmocr colab

* modified mmocr colab

* modified mmocr colab

* modified mmocr colab

* modified mmocr colab

* modified colab link

* rerun pre-commit

* rerun pre-commit
2021-05-18 11:13:07 +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
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
Hongbin Sun 05aedf11d6
update demo docs for batch inference (#181) 2021-05-13 17:58:38 +08:00
quincylin1 7e0d48df11
fix metafiles (#182) 2021-05-13 17:28:59 +08:00
Hongbin Sun 97e2f27017
fix #173: support aug test (#178)
* fix #173: support aug test

* fix pytest

* support batch inference for both text det and recog

* update unittest

* use one img for batch infer test
2021-05-13 07:18:00 +00:00
quincylin1 8e65bea9c3
Add mminstall (#180)
* add mminstall.txt

* fix mminstall.txt

* fix mminstall
2021-05-13 15:06:12 +08:00
quincylin1 4b17d43bb0
add manifest.in (#175) 2021-05-13 03:16:10 +00: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
quincylin1 ee700e94cd
add mminstall.txt (#177) 2021-05-13 09:01:31 +08:00
quincylin1 ca1ddfd6fd
add include_package_data to setup.py (#176) 2021-05-12 22:55:50 +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
lizz 9be85faa82
Add things from "import *" to __all__ (#167)
* Add things from "import *" to __all__

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

* Remove tile_image

Signed-off-by: lizz <lizz@sensetime.com>
2021-05-12 20:22:30 +08:00
lizz 04273b325a
Update image link (#170)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-12 06:16:22 +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
Sergio Bugallo Enjamio 18c54afbdc
Add support for batch inference (#86)
* Add support for numpy arrays in model_inference

* Add test for numpy ndarray inference

* Fix linting problems

* Add support for batch inference

* Add batch inference demo script

* Fix comment

* Test batch inference with paths and arrays

* lint code

* Update model_inference docstring

* Refactor model inference tests

* Change inference function to make text detectors and recognizers use the same input data types

* Change single state text detector model to support batch inference

* Lint code

* simplify inference tests

* Remove psenet from batch inference test cases to prevent the pytest being killed

* Update batch_image_demo.py

* fix bug when test with dataset 

fix bug when test with dataset, for example, `./tools/dist_test.sh configs/textrecog/sar/sar_r31_parallel_decoder_academic.py <checkpoint> 1 --eval acc`

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-12 02:04:01 +00: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
lizz 6a1d1020f6
Add a message when annotation is empty (#165)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-11 16:38:30 +08:00
Hongbin Sun 85b821ab79
fix bug of loss nan (#159)
* fix bug of loss nan

* update

* update ctc loss
2021-05-11 16:13:56 +08: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
jeffreykuang 47896a3f80
Unit tests of dbnet transform (#163)
* Add unit test for dbnet transforms

* Add unit test for dbnet transforms

* Add unit test for dbnet transforms
2021-05-11 15:21:14 +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