Commit Graph

249 Commits (1a379f2f1b9c89b4c927b70081a170579df47559)

Author SHA1 Message Date
Tong Gao 3059c97dc5
[Docs] Configure Myst-parser to parse anchor tag (#1012) 2022-05-12 22:49:55 +08:00
Xinyu Wang 9cfa29f862
[Docs] Fix typos (#1001) 2022-05-06 17:15:36 +08:00
Tong Gao 1962c249a6
Bump version to 0.6.0 (#996)
* initial

* update changelog

* update version num

* fix

* fix letter case

* grammar
2022-05-05 22:16:41 +08:00
Qing Jiang de2851e3c2
[Feature] Add HierText converter (#948)
* loss

* fix

* [feature] add hiertext

* fix name

* update docs

* update

* update markdown

* update doc

* update doc

* update docs
2022-05-05 16:31:36 +08:00
Jianquan Ye 061d353b2a
[Model] Add MASTER (#807)
* fix #794: add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* Fix linting

* after git rebase main

* after git rebase main

* fix conflict add MASTER

* fix conflict add MASTER

* after git rebase main

* fix conflict add MASTER

* fix conflict add MASTER

* fix conflict add MASTER

* after git rebase main

* add GCAModule to plugins

* coexist master and master_old

* fix merge mmocr 0.5.0 conflict

* fix lint error

* update

* [fix] remove remains in __init__

* [update] update code in review

* update readme for master

* Add docstr to MasterDecoder, refined MasterDecoder, remove MASTERLoss

* Unify the output length of MasterDecoder in train and test mode; add test for it, remove MasterLoss

* update readme

* update

* update metafile,README,demo/README,config,ocr.py

* Update mmocr/utils/ocr.py

* update

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
Co-authored-by: Mountchicken <mountchicken@outlook.com>
2022-05-05 16:06:15 +08:00
Tong Gao 137713fbc3
[Docs] Renew dataset_types.md (#997)
* refactor dataset_types

* fix

* add docs for evaluation parameters in textdet & recog datasets

* HardDiskLoader -> AnnFileLoader
2022-05-05 15:22:15 +08:00
Hongbin Sun a2d741b8a7
[Feature] Add labelme converter for textdet and textrecog (#972)
* add labelme converter

* move to common

* add labelme sample annos

* add doc

* remove useless field generated by labelme to reduce size

* add recog_format option; add skip ignored instances while cropping

* set warp as false by default

* update doc

* fix typo

Co-authored-by: xinke-wang <wangxinyu2017@gmail.com>
Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>
2022-05-03 17:28:22 +08:00
Qing Jiang 92ef554a82
[Feature] Add recog2lmdb and new toy dataset files (#979)
* loss

* fix

* add img2lmdb and test files

* update

* add reference

* fix lint

* fix typo

* use total_numer instead to fit mmocr's lmdbloader

* reorganize and update

* fix lint

* update test file

* refactor and update

* fix test

* update doc in tools

* fix lint

* update old lmdb test file

* update

* mask the unittest for recog2lmdb and use json format for label_only

* remove if __name__

* fix case, doc, typo, formats

* fix typos

* fix docs and variable names

* Apply suggestions from code review

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>

* update test_loader.py and fix a bug

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>
2022-04-29 22:30:36 +08:00
Qing Jiang d8549ade85
[Feature] Support loading images and labels from lmdb file (#982)
* loss

* fix

* add LoadImageFromLMDB

* update

* new version

* add json

* fix bug

* Compatible with older versions

* modify test_loader

* add docstring and update test_loader

* fix a bug

* fix docstring and update doc

* update

* Update mmocr/datasets/utils/backend.py

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>

* Update docs/en/tutorials/dataset_types.md

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>

* fix lint

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>
2022-04-29 15:46:36 +08:00
Tong Gao 64f02ba1a2
[Fix] Update installation step of `albumentations` (#984)
* [Fix] Update installation step of `albumentations`

* update albu installation guide
2022-04-28 00:01:27 +08:00
Qing Jiang 24ac18e653
[Update] Update the version requirement of mmdet in docker (#966)
* loss

* fix

* update

* update mmdet version in doc

* update

* update __init__
2022-04-24 10:12:39 +08:00
Xinyu Wang 06b73cf71a
[Fix] Fix TotalText Anno version issue (#945)
* fix tt converter version issue; fix typos in docs

* remove incorrect descriptions

* fix docstring & incorrect file name

* fix docstring identation
2022-04-23 23:57:21 +08:00
Tong Gao c08c113bd4
[Enhancement] Support latest MMCV (#959) 2022-04-21 19:10:52 +08:00
Xinyu Wang 9c54e7eb00
[Feature] Add RCTW dataset converter (#914)
* add rctw

* fix typos
2022-04-18 09:27:18 +08:00
Xinyu Wang 20fc909fc4
[Feature] Add LSVT Data Converter (#896)
* add lsvt

* fix name

* fix name

* update

* add lsvt

* set default val 0

* fix a bug

* fix typos

* fix file name

* fix lint

* fix lint
2022-04-18 09:15:42 +08:00
Xinyu Wang c155935087
[Docs] update recog.md for lmdb generation [ci skip] (#934) 2022-04-17 09:02:24 +08:00
Y-M-Y 3179e4d6bd
[Feature] Add analyze_logs in tools and its description in docs (#899)
* Create analyze_logs.py

* Update tools.md

* fix lint and typo

* Update analyze_logs.py

* Add arg table and demo log file

* Delete line66 for lint error

* fix captial letters

* update doc

* fix markdown indentation

* Add log_analysis_demo.png to demo/resources

* Add log_analysis_demo.png and two links in table

* Improve epoch-based metric

* fix lint error

* fix lint error(tabs and spaces)

* check code lints and format

Co-authored-by: Mountchicken <mountchicken@outlook.com>
Co-authored-by: xinke-wang <wangxinyu2017@gmail.com>
2022-04-02 22:40:39 +08:00
Tong Gao 4c57bd35ac
[Docs] Fix the misleading description in test.py (#908)
* [Docs] Fix misleading description in test.py

* update docs
2022-04-02 11:31:48 +08:00
Tong Gao 0546134771
Bump version to 0.5.0 (#904) 2022-03-31 17:42:24 +08:00
Tong Gao 7a6b9848ed
[Docs] Reorganize the directory structure section in det.md (#894)
* [Docs] Reorganize the directory structure section in det.md

* improve

* fix indentation

* Fix structure

* sync dataset order to the overview

* format det doc

* fix |
2022-03-31 15:10:21 +08:00
liukuikun 581c9affba
[Enchance] Recog dataset doc (#893)
* [Enhancement] refactor recog datset content tree

* update recog

* update
2022-03-31 15:10:02 +08:00
Qing Jiang 1ddfd7f36e
[Docs] update doc for ILST, VinText, BID (#902)
* loss

* fix

* update
2022-03-30 22:37:38 +08:00
Xinyu Wang c6bb105b83
[Docs] Update Instructions for New Data Converters (#900)
* update docs

* fix spaces & add deprecation

* fix funsd

* remove repeated docs
2022-03-30 22:07:17 +08:00
Xinyu Wang ee2c3cfd46
[Feature] Add DeText Converter (#818)
* add DeText Converter

* Update tools/data/textrecog/detext_converter.py

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

* update doc; support jsonl; fix docstrings

* update mkdir func

* fix bug

* update doc; do not filter for test val

* move directory tree

* fix indentation

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-30 14:43:33 +08:00
Xinyu Wang 8b928cb500
[Feature] Add NAF Converter (#815)
* NAF dataset downloading command

* add NAF converter

* revert incorrect url revision

* fix typo

* support jsonl; save filtered crops; add data description in docstring; update ddoc

* remove preserve-symbol; update docs; fix special symbol filter

* move tree structure

* fix indentation

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-03-30 14:31:47 +08:00
Xinyu Wang bdd32c8052
[Feature] Add SROIE Converter (#810)
* add SROIE converter

* add sroie converter

* fix docstring indentation

* fix lint

* remove val split; add test split

* delete google drive timestamp

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

* remove timestamp

* update docs; support jsonl; fix crop

* move tree structure

* move tree structure

* move directory tree

* fix indentation

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-30 13:14:23 +08:00
Xinyu Wang 958e4a3e87
[Feature] Add LV Dataset Converter (#871)
* add LV converter

* add docs

* add recog converter; update doc
2022-03-29 11:50:27 +08:00
Tong Gao b37b1d05dd
[Fix] Use hyphen instead of underscores in script args (#890) 2022-03-29 10:28:38 +08:00
Xinyu Wang 33c5e41724
[Enhancement] Add warning info for LineStrParser (#850)
* Add warning info for LineStrParser while more than two blanks were detected.

* add LineJsonParser Doc

* fix typo

* fix typo

* fix typo

* separate tutorial section
2022-03-22 17:44:05 +08:00
Tong Gao 838aa47f9f
[Feature] Add args and docs for multi-machine training/testing (#849)
* add args and docs for multi-machine training/testing

* update docs

* update docs

* update docs

* update docs

* revert commit
2022-03-22 16:25:39 +08:00
JiangQing 44ee889141
[Doc] Update docs about installation on Windows (#852)
* support windows

* add macOS
2022-03-18 13:23:22 +08:00
shaohua.zhang f663dce26f
[Docs] Add the model serving translation document (#845)
* Update README_zh-CN.md

* Update README_zh-CN.md

* Add files via upload

* Update getting_started.md

* Update getting_started.md

* Update getting_started.md

* Add files via upload

* Update model_serving.md

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-17 19:53:13 +08:00
shaohua.zhang 3d177deae9
[Docs] Add getting_started.md in docs/zh (#841)
* Update README_zh-CN.md

* Update README_zh-CN.md

* Add files via upload

* Update getting_started.md

* Update getting_started.md

* Update getting_started.md

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-16 14:18:13 +08:00
Xinyu Wang 368325f129
[Docs] Fix confused description for load-from and resume-from (#842)
* fix confusions

* fix confusions
2022-03-15 14:19:33 +08:00
kira b0101839de
[Docs] Translate testing.md. (#822)
* Create testing.md

* Update testing.md

* Update testing.md
2022-03-11 20:38:20 +08:00
Tong Gao c25404b358
[Enhancement] Option for different seeds on different ranks (#820) 2022-03-10 10:43:36 +08:00
Tong Gao f1609b50e9
[Docs] Correct misleading section title in training.md (#819)
* [Docs] Correct misleading section title in training.md

* grammar
2022-03-09 16:52:43 +08:00
Xinyu Wang 14c75da7bd
[Feature] Add FUNSD Converter (#808)
* Add FUNSD Converter

* Update tools/data/textrecog/funsd_converter.py

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

* Update tools/data/textrecog/funsd_converter.py

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

* Update tools/data/textdet/funsd_converter.py

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

* blank line between sections

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

* fix incorrect docstrings

* fix docstrings & fix timer

* add --preserve-vertical arg for preserving vertical texts

* fix --preserve-vertical

* [doc] fix recog.md incorrect description

* fix docstring style

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

* fix docstring spaces

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-04 12:25:54 +08:00
Tong Gao ac4462f374
[Feature] Add CurvedSyntext150k Converter (#719)
* [Feature] Add bezier_to_polygon to box_util

* Add num_sample to parameter

* add sort_point util

* update docstring

* Add curvedsyntext converter
2022-03-02 11:02:14 +08:00
DEEPBLUE 186aecbf59
[Docs] Add demo/README_zh.md (#802)
* add README_zh.md

* translate

* rough translation

* refine demo/README_zh.md

* fixes #802

* fix #802

* add README_zh.md

* fixes #802

* Update README_zh.md

* fix white-space problem

* Update demo/README_zh.md

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

* Update demo/README_zh.md

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

* fix Readthedocs' keywords @gaotongxiao

* Fix minor errors, rename the file and process this file in the scripts for redthedocs

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-03-02 09:51:37 +08:00
Tong Gao b91421e8b1
[Docs] Add detailed version requirement tables (#778) 2022-02-16 20:42:31 +08:00
liukuikun 1aae45b7d3
[Docs] Relax mmcv version requirement (#775)
* update ci and readme

* Update docs/en/install.md

* Update docs/zh_cn/install.md

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-02-10 09:41:36 +08:00
Tong Gao a75fc6bf99
Bump version to 0.4.1 (#755)
* 0.4.1 change log

* update version number

* update changelog

* update changelog

* update dates

* update version
2022-01-27 14:39:17 +08:00
liukuikun e0ff11819b
[Feat] Support specify gpu for testing and training with gpu-id instead of gpu-ids and gpus (#756)
* support set gpu number for testing and deprecate gpu_ids and gpus with gpu_id

* update gpu-ids gpus and gpu-id dostring

* add args.gpus process

* fix typo

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-01-27 10:11:14 +08:00
Tong Gao a809a52c78
[Feature] Support CPU training/testing (#752)
* [Feature] Support CPU training/testing

* check mmcv version for cpu training
2022-01-27 09:59:09 +08:00
liukuikun 976b976e4a
[Docs] update readme according to standard (#742)
* config/readme standard

* add ABINet linking

* fix linking error, delete unused cite and adjust note sytle

* union note format

* Remove > before paper link when generating docs
2022-01-25 23:02:59 +08:00
Tong Gao 94a1de4fdb
[Docs] Update docs link (#710) 2022-01-05 21:44:13 +08:00
Tong Gao a0f1c011c0
[Docs] Add model summary (#693)
* Add model summary

* Update text recog jpg

* update pictures and add kie
2021-12-31 11:26:38 +08:00
Tong Gao 96beda6f2d
[Docs] Add config tutorial (#683)
* Add config tutorial and reorganize dataset_type.md

* update cn docs toc

* Add misc tool

* Improve docs api menu and config docs
2021-12-28 12:41:24 +08:00
Tong Gao 22a7d384dd
[Docs] Fix recog.md (#674) 2021-12-15 07:04:41 +00:00
Tong Gao af9a62502c
Bump version to 0.4.0 (#672) 2021-12-15 11:34:56 +08:00
Tong Gao 9104667112
[Model] Full ABINet Framework (#651)
Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2021-12-15 11:21:54 +08:00
Hongbin Sun a4237ad568
[Enhancement] Support muitiple pipelines for different datasets (#657)
* support multi pipelines

* support different pipelines

* fix

* fix replace_image_to_tensor

* remove redundant check

* update

* add copyright

* not in-place modify
2021-12-14 12:29:43 +00:00
Tong Gao d58fac9760
[Docs] Docs fix (#664)
* [Docs] Update readme links

* Fix docs merge script
2021-12-13 19:08:42 +08:00
Tong Gao b04775fd78
[Docs] Refactor docs structure (#662)
* Refactor docs structure

* update script paths

* update path ignore
2021-12-10 17:19:27 +08:00
Tong Gao 6d1a5ce046
[Docs] Use shared menu from theme instead (#655) 2021-12-10 11:39:21 +08:00
Tong Gao 925b365dcf
[Docs] Add MMHuman3D in readme (#644)
* [Docs] Add MMHuman3D in readme

* add docs
2021-12-03 16:52:38 +08:00
Tong Gao e203d546b4
[Docs] Docstring for textdet models (#561)
* add docstring for dbnet
2021-12-02 18:57:01 +08:00
Tong Gao 69d5040590
[Docs] Update model readme (#604)
* update satrn

* Add intro and pic to readmes

* update stats.py

* abstract for tps

* Fix CRNN and TPS CRNN readme
2021-11-30 16:40:18 +08:00
Tong Gao 04ef6053a6
[Docs] Add MMFewShot (#621) 2021-11-26 11:18:57 +08:00
Tong Gao 53b050b0ca
[Docs] Add MMFlow & MIM (#597)
* [Docs] Add MMFlow

* add mim
2021-11-19 14:16:42 +08:00
apiaccess21 284b5acc12
[Docs] C++ example section for deployment (#593)
* C++ example section

* Optimize the section for readthedocs

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-17 14:16:54 +08:00
Tong Gao 9de4ddf952
fix broken links (#576) 2021-11-11 17:57:15 +08:00
Tong Gao 34bd6b9bbb
[Docs] Docs for KIE CloseSet & OpenSet (#573)
* add kie docs

* fix paths

* sync cn docs

* fix layout error
2021-11-11 17:39:27 +08:00
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
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
Hongbin Sun 4eccb1cfcb fix #13: link pr to issue 2021-04-04 13:01:21 +08:00
Hongbin Sun a347a97c23 fix #13: update docs with toc 2021-04-04 11:56:14 +08:00
Hongbin Sun dd120271ba
fix #11: update docs (#12)
* fix #11: update docs

* fix #11: update datasets.md for kie
2021-04-04 00:19:55 +08:00
Hongbin Sun 3c0f9d87c5
fix #9: [feature]: setting norms for contributing (#10)
* fix #9: [feature]: setting norms for contributing

* fix #9: [feature]: setting norms for contributing

* fix #9: [feature]: setting norms for contributing

* fix #9: [feature]: setting norms for contributing
2021-04-03 19:40:29 +08:00
quincylin1 8476c239b2 [feature]: dbnet and docs 2021-04-03 01:21:33 +08:00
yuexy 72135ed552
Revert "documentation and dbnet related code" 2021-04-03 01:00:26 +08:00
quincylin1 b031934129 documentation and dbnet related code 2021-04-03 00:41:23 +08:00