Commit Graph

85 Commits (a50b0c9fb9196fbd5208400a32ec0bbc068d3e8b)

Author SHA1 Message Date
Tong Gao 1efdbda790
[Docs] Update docs menu (#564)
* update docs menu

* fix wrong links
2021-11-09 17:16:13 +08:00
Darwin Bautista 80741e1479
[Feature] Add converter for the Open Images v5 text annotations by Krylov et al. (#497)
* Add converter for the OpenVINO annotations for Open Images by Krylov et al.

Open Images V5 Text Annotation and Yet Another Mask Text Spotter
Paper: https://arxiv.org/abs/2106.12326

* docs fix & add chinese docs
2021-10-28 16:49:36 +08:00
Tong Gao 027cb7aac5
[Docs] TorchServe docs (#539)
* Add torchserve docs

* polish contents and fix the visiualization bug in torchserve testing

* polish content
2021-10-22 16:49:28 +08:00
liukuikun 5671939bbc
[Fix] fix converting to onnx bug because of changing key from img_shape to resize_shape (#523)
* fix pytorch to onnx for mmocr0.3.0

* fix pytorch to onnx for mmocr0.3.0

* modify deployment doc and fix scale_factor bug and test psenet and panet

* fix trt part

* init detector with **(cfg.model)

Co-authored-by: AllentDan <AllentDan@yeah.net>
2021-10-20 19:55:57 +08:00
Tong Gao 75d32504e0
[Feature] Add codespell pre-commit hook and fix typos (#520)
* add spellcheck hook

* fix typos
2021-10-13 22:56:11 +08:00
Tong Gao f72012092f
[Docs] Enhancement & sync Chinese docs (#492)
* reorder navigation links, add links to twitter and zhihu

* update chinese docs
2021-09-28 11:32:44 +08:00
Tong Gao e23b765956
[Docs] Improve docs style (#474)
* new theme

* add theme

* update zh_cn

* improve docs style

* use customized

* fix

* update req

* docs

* Update docs

* update conf

* update

* update layout

* disable logo url

* free version limit

* update conf

* Fix api ref

* fix version
2021-09-08 11:40:51 +08:00
Tong Gao f9d158fafa
Bump version to 0.3.0 (#461)
* Bump version to 0.3.0

* update changelog
2021-08-25 16:43:31 +08:00
Tong Gao c0728c49b8
[Docs] Refactor docs (#409) 2021-08-25 16:41:07 +08:00
LvTianlei 8f377f2fa2
[Feature] Add Satrn (#405)
* Add SATRN

* Create satrn_small_academic.py

* Update README.md

* change config name

* Update mmocr/models/textrecog/backbones/shallow_cnn.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_academic.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_small.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update shallow_cnn.py

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update test_ocr_encoder.py

* change keep_aspect_ratio=False

* Update transformer_layer.py

* Update configs/textrecog/satrn/satrn_small.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update configs/textrecog/satrn/satrn_academic.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update mmocr/models/textrecog/layers/transformer_layer.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update transformer_layer.py

* Apply suggestions from code review

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>

* Update transformer_layer.py

* update satrn readme

* add satrn to ocr.py

* add satrn_sm and fix configs

* add a test for config

* add copyright info

* use mmocr registry

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-08-19 22:02:58 +08:00
Tong Gao b8f7ead74c
[Enhancement] Add copyright info (#439)
* add copyright info
2021-08-17 17:39:30 +08:00
Tong Gao 304b48e187
[Docs] Full screen support in readthedocs (#413)
* add full screen css

* New dir structure
2021-08-09 11:16:43 +00:00
Tong Gao 2cbb1a6cd6
[Docs] Fix docs (#395)
1. Supplement links for ct80 and svtp
2. Simplify getting_started.md
3. clarify batch_mode support in demo.md
2021-08-04 14:26:16 +08:00
Tong Gao 7b16754006
[Docs] Fix API reference & version, support exporting pdf & epub (#406)
* add mmdet

* remove uncessary autodoc mock import

* update docs version

* bind docs version to version.py

* support pdf and epub

* fix readthedocs
2021-08-04 14:24:52 +08:00
Omkar Manjrekar 68ec3f5519
Add kie image demo and docs. (#374)
* Add kie_image_demo.py

* Add kie demo docs

* Add brief instructions for kie image demo

* Add ann file field and return data for kie image demo

* Follow lint and import rules

* Fix bugs, reuse functions in KIEDataset, and use a new demo pic

* Add config-dir and fix indexing bug in ocr script

* [Feature] Improve ocr.py

1. Add box stitching back to ocr.py
2. Add config_dir which allows users to specify the default config path
3. Warn users when overriding parameters are set
4. Allow users to use customized checkpoint files

* Add docs for new ocr.py

* Add docs for merge

* Support kie in ocr.py

* Merged kie to ocr.py

* update docs, remove unsupported unvisual sdmgr

* Update mmocr/apis/inference.py

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>

* Apply suggestions from code review

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>

* fix linting

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-08-04 14:20:13 +08:00
Tong Gao 9c76f7b5cc
[Feature] Improve ocr.py (#400)
* [Feature] Improve ocr.py

1. Add box stitching back to ocr.py
2. Add config_dir which allows users to specify the default config path
3. Warn users when overriding parameters are set
4. Allow users to use customized checkpoint files
5. Add docs
2021-08-03 15:57:30 +08:00
SamAyala a2af2b4d31
Add output option to det_recog_show_result method (#386)
* add out_file arg

* fix small bug

* Update visualize.py
2021-07-25 12:29:40 +08:00
SamAyala 76785e185a
User friendly API v2 + Docs ! (#371)
* major update

- Refactor code
- Support for folder and list/tuple or np.arrays or img paths
- Better export method
* feature update

- Batch size support
- More refactoring

* added docs


* Optimize docs structure, fix improper layout in readthedocs


Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-07-25 00:22:27 +08:00
Tong Gao 02d657d141
[Fix] upgrade version requirement of mmdet to 2.14.0 to avoid known bugs (#382) 2021-07-21 18:01:55 +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 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
Tong Gao 0effc39c5f
[Docs] fix dead links, remove invalid TOC in CN docs (#367) 2021-07-15 08:28:36 +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 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
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
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