0f041d4250
[Fix] updata LabelData import
2022-07-21 10:50:57 +08:00
f6b72b244b
Add ImgAug and tests
2022-07-21 10:50:57 +08:00
cb85f857aa
Add BaseTextDetPostProcessor
2022-07-21 10:50:57 +08:00
f0c6d44ce8
Add dump_ocr_data
2022-07-21 10:50:56 +08:00
98bc90bd1c
[Feature] TextRecogSample
2022-07-21 10:50:56 +08:00
c920edfb3a
[Refactor] split labelconverter to Dictionary
2022-07-21 10:50:56 +08:00
c47c5711c1
[Feature] TextDetSample
2022-07-21 10:50:56 +08:00
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
liukuikun
gaotongxiao
wangxinyu.vendor
Jianquan Ye
Xinyu Wang
Hongbin Sun
Qing Jiang
garvan2021
liukuikun
yuexy
Jianyong Chen
mpena-vina