41c1671e7b
Refactor PyramidRescale
2022-07-21 10:50:56 +08:00
b5fc589320
Add CI
2022-07-21 10:50:55 +08:00
69e6c80558
move tests to old_tests, add empty test folders
2022-07-21 10:50:55 +08:00
1ad4caa7cb
[Enhancement] Allow choosing metrics to report in text recognition tasks ( #989 )
...
* [Enhancement] Allow choosing metrics to report in text recognition tasks
* add acc support
* refine docstr
2022-05-05 16:10:38 +08:00
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
fbc138d2d3
[Feature] Add DBNet++ ( #973 )
...
* add dbnet++
* fix docstring
* fix some param names
* fix
* fix docstring
* add init
* add doc; remove configs
* add dbnet++ to readme
* fix readme
* update config
* update readme
* update readme
* update ocr.py
* update metafile.yml
* update readme
* update readme
* move to dbnetpp
* fix paths
* fix head level
* fix typo
* update demo.md
* Update configs/textdet/dbnetpp/README.md
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* fix typo
* fix link
2022-05-05 14:39:20 +08:00
6b180db93d
[Enhancement] Automatically report mean scores when applicable ( #995 )
2022-05-05 12:57:19 +08:00
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
c91763ad09
[Fix] update label.lmdb ( #991 )
...
* loss
* fix
* hotfix
2022-04-29 23:26:32 +08:00
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
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
064a2b8c45
[Feature] Support computing mean scores in UniformConcatDataset ( #981 )
...
* Get avg results in UniformConcatDataset
* add docstr
* Fix
* fix test
* fix typo
2022-04-29 14:48:21 +08:00
0c8fa52b22
[Fix] Fix ImgAug transform ( #949 )
...
* [Fix] Fix bbox case in ImgAug
* Finialize the augmentation and add tests
2022-04-28 22:52:11 +08:00
888f700134
[Enhancement] Support dynamic threshold range in eval_hmean ( #962 )
...
* [Enhancement] Support dynamic threshold range in eval_hmean
* upgrade textdetdataset, add deprecate warning
2022-04-22 17:07:36 +08:00
b11fa80d82
[Fix] correct meta key ( #926 )
...
* correct meta key
* fix test metakey
2022-04-18 09:09:09 +08:00
c79a62487d
[Feature] Add Tesserocr Inference ( #814 )
...
* append tesserocr to requirements list, but may encounter build error at windows platform
* simply save
* 2022.3.4
* opencv-python==4.5.5 can cause cv2.error when print_result=True
* append MMOCR.tesseract_det_inference()
* argument check append
* fix lint error
* update commentary
* lint fix
* requirement remove opencv
* handle tessdata problem
* support tesseract recognition
* fix some bugs
* fix imshow bug
* support batch mode(fake)
* modify annotation
* refactor BaseRecognizer for show_result
* append pytest
* Mock tesseract
* Fix test
* remove \n from Tesseract
* normalize text score
* update docs
2022-03-28 22:50:53 +08:00
37833ada4b
[Fix] Support loading space character from dict file ( #854 )
...
* [Feature] Support loading dict file with space character
* fix tests
* clean up \\r in tests
* add DICT37 and DICT91
* update docstr
2022-03-22 17:44:32 +08:00
72f9454573
[Refactor] Refactor ResNets ( #809 )
2022-03-18 10:36:03 +08:00
680dff373e
[Feature] Support jsonl in recognition converter ( #844 )
2022-03-18 09:22:32 +08:00
402e8f1162
[Feature] Support loading annotation file with petrel/http backend ( #793 )
...
* add petrel backend for ann_file loading
* support icdar dataset
* add http backend
* name back
* fix pytest
* support load lmdb file on ceph
* refactor
* add close for lmdb
* add log and warning msg
* rename
* typo
* update
* fix
2022-03-07 19:18:02 +08:00
fb77352eb2
[Fix] Standardize the type of torch.device in ocr.py ( #800 )
2022-03-03 14:18:33 +08:00
3110ab7863
[Enhancement] Add windows CI ( #790 )
...
* [Enhancement] Add windows CI
* [Enhancement] Add windows CI
* update
* update
* update
* [Fix] using assert will keep lmdb file opend and fail to cleanup in test_loader.py
* [Fix] map size should be small on windows in lmdb_util.py
* [Fix] Fix some bugs
* [Fix] Fix some bugs
* [Fix] Fix some bugs
* remove comments & fix bugs
Co-authored-by: Mountchicken <mountchicken@outlook.com>
2022-03-02 10:34:15 +08:00
5582e17feb
[Enhancement] Automatically infer device for inference if not speicifed ( #781 )
...
* [Enhancement] Automatically infer device for inference if not speicifed
* update docs
* fix unit test
2022-02-16 20:42:47 +08:00
0f5c7d38f8
[Fix] Disable worker in the dataloader of gpu unit test
2022-02-16 20:42:22 +08:00
c466e6f82d
[Enhancement] Upgrade isort ( #771 )
...
* [Enhancement] Upgrade isort
* Fix files
2022-02-09 23:22:14 +08:00
d871eed79e
[Enhancement] Add setup multi-processing both in train and test. ( #757 )
2022-01-26 20:34:19 +08:00
82488024db
[Feature] Add bezier_to_polygon and sort_points to box_util ( #703 )
2022-01-17 15:26:55 +08:00
fb1892a1ae
[Fix] fix bug of show_results in single_gpu_test ( #667 )
...
* move single_gpu_test to mmocr/apis
* fix show result in single_gpu_test
* add unittest
* add docstring
* add unittest for kie
* avoid re-define
* remove skipif
* add skip
* fix conflict
* fix with comment
* add note in readme
* update
2021-12-23 03:30:01 +00:00
f8dfbd4177
[Fix] Support both ConcatDataset and UniformConcatDataset ( #675 )
...
* support UniformConcatDataset
* update
* rm useless
* handle 2d-list datasets
2021-12-22 12:32:02 +00:00
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
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
7874531db3
[Refactor] refactor textdet postprocess ( #640 )
...
* refactor textdet postprocess using build_postprocessor
* update docstring
* update docstring
* update postprocess docstring
* compatible with previous configs
* rename to BaseTextDetector
* rename to DRRGPostprocessor
* fix
* update warning
* fix
* fix conflict
* rename back
* rename warpper to utils
* fix doc
2021-12-14 17:35:44 +08:00
5caa945a8d
fix #614 : textsnake targets ( #660 )
...
* fix #614 : textsnake targets
* fix lint
* add textsnake_targets test cases
* init with eps
* fix test coverage
2021-12-13 17:45:30 +08:00
c6c4acd39b
[Fix] fix type check ( #650 )
...
* fix check type
* add notice
2021-12-09 12:45:59 +08:00
fec986aba0
[Enhancement] Use shape from loaded image ( #652 )
...
* use shape from loaded image
* complete unittest
* add params as gate
* add comment
* add assert
2021-12-09 01:34:09 +00:00
0a1787d6bc
[Refactor] refactor transformer modules ( #618 )
...
* base refactor
* update config
* modify implementation of nrtr
* add config file
* add mask
* add operation order
* fix contiguous bug
* update config
* fix pytest
* fix pytest
* update readme
* update readme and metafile
* update docstring
* fix norm cfg and dict size
* rm useless
* use mmocr builder instead
* update pytest
* update
* remove useless
* fix ckpt name
* fix path
* include all config file into pytest
* update inference
* Update test_recog_config.py
2021-12-04 17:12:31 +08:00
5a8859fe66
[Refactor] Refactor text detection config ( #626 )
...
* refactor textdet configs
* remove duplicate keys in _base_
* remove import from config
* syncbn to bn on cpu
* minimize change
2021-12-03 19:37:43 +08:00
d3d56182c6
fix #630 : Add input size check to model_inference ( #633 )
...
* fix #630 : Add input size check to model_inference
* fix #630 : Added exception when receiving empty image
2021-12-02 03:00:23 +00:00
809661ea28
[Enhancement] CircleCI Setup ( #611 )
...
* Add .circleci/config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* fix unit test
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
* Updated config.yml
2021-11-23 19:54:38 +08:00
a50b0c9fb9
[Feature] Support openset kie ( #498 )
...
* add openset kie dataset
* updare readme
* add anno convert script
* update docstring
* update script
* add & update docstring
* fix typo
* update docstring format
2021-11-11 14:47:38 +08:00
9f42d78db7
[Feature] Save filename in text detection test results ( #570 )
...
* save filename in textdet test results
* fix pytest
2021-11-11 14:07:07 +08:00
aa4b5b5574
[Fix] Allow invalid polygons in intersection and union by default ( #471 )
...
* Allow invalid polygons in intersection and union by default
* update docs
2021-09-28 10:57:12 +08:00
0881c2d2a2
[Fix] Add zero division handler in poly utils, remove Polygon3 ( #448 )
...
* Add check to avoid zero div in iou computation
* replace polygon3 with shapely
* remove req of Polygon3
2021-08-25 13:14:58 +08:00
7c1bf45c63
[Fix] fix inconsisent training/eval state after SyncBN->BN ( #453 )
2021-08-25 13:14:03 +08:00
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
7571763376
[Refactor] Use MMOCR's registry ( #436 )
...
* [Refactor] Use MMOCR's registry
1. Define MMOCR's registries as a child of MMDet's
2. Register all models to MMOCR's own registries
3. Modify some model configs so that some models in MMDet can be
correctly located
4. Remove some outdated demo scripts
* add detectors
2021-08-19 19:17:15 +08:00
b8f7ead74c
[Enhancement] Add copyright info ( #439 )
...
* add copyright info
2021-08-17 17:39:30 +08:00
9b5b25ef71
[Fix] fix a bug of DBNet when text repr type is poly ( #421 )
...
* fix dbnet bug when text repr type is poly
* add db_decode unit test
2021-08-13 21:39:53 +08:00
80a0536c7c
[Enhancement] Add tests for ocr.py ( #428 )
...
* minor fix to ocr.py
* add test for ocr.py
2021-08-13 09:55:33 +08:00
906faec372
update digit_version and add a test ( #427 )
2021-08-12 08:53:45 +00:00
wangxinyu.vendor
gaotongxiao
Jianquan Ye
Xinyu Wang
Hongbin Sun
Qing Jiang
garvan2021
liukuikun
yuexy
Jianyong Chen
mpena-vina
LvTianlei
liukuikun