Commit Graph

60 Commits (2a2cab3c8c17cac31e6d00f2b68f9eb467a15727)

Author SHA1 Message Date
Tong Gao ad470e323a
[Feature] Refactor Inferencers (#1608)
* tmp commit

* initial

* kie

* update MMOCRInferencer and ocr.py

* fix

* fix bug & add ut

* ut for kie

* part of mmocr inferencer ut

* part of mmocr inferencer ut

* ut

* ut

* docs

* inferencer

* Add TextSpotInferencer

* test

* fix

* textspot

* fix

* test

* test

* fix

* fix
2023-02-02 19:05:55 +08:00
Tong Gao a82fc66812
[Feature] Add RepeatAugSampler (#1678)
* [Feature] Add RepeatAugSampler

* fix
2023-01-31 19:42:48 +08:00
liukuikun 689ecf0f5f
fix LoadOCRAnnotation ut (#1695) 2023-01-31 11:23:49 +08:00
Tong Gao 6992923768
[Enhancement] Discard deprecated lmdb dataset format and only support img+label now (#1681)
* [Enhance] Discard deprecated lmdb dataset format and only support img+label now

* rename

* update

* add ut

* updata document

* update docs

* update test

* update test

* Update dataset.md

Co-authored-by: liukuikun <641417025@qq.com>
2023-01-17 10:10:51 +08:00
Tong Gao b0557c2c55
[Transforms] SVTR transforms (#1645)
* rec transforms

* fix

* ut

* update docs

* fix

* new name

* fix
2023-01-06 16:04:20 +08:00
liukuikun 9baf440d7a
[Feature] ConditionApply (#1646) 2022-12-28 11:53:32 +08:00
Tong Gao 101f2b6eef
[Enhancement] Enhance FixInvalidPolygon, add RemoveIgnored transform (#1632)
* fix polygon_utils

* ut for poly_make_valid

* optimize crop_polygon

* FixInvalidPolygon, debug msg included

* add remove_pipeline_elements to utils

* enhance fixinvalidpolys

* fix transform_utils

* remove ignored

* RemoveIgnored

* add tests

* fix

* fix ut

* fix ut
2022-12-27 10:30:10 +08:00
liukuikun ffe5237aa8
[Refactor] Refactor icdardataset metainfo to lowercase. (#1620) 2022-12-13 11:45:33 +08:00
liukuikun b8c445b04f
[Fix] fix icdar data parse for text containing seperator (#1587)
* [Fix] fix icdar data parse for text containing seperator

* Update mmocr/datasets/preparers/parsers/base.py

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-12-01 18:43:09 +08:00
Tong Gao cfce57ad87
[Feature] Add config generator (#1552)
* [Feature] Add config generator

* update icdar2013

* fix ut

* simplify design

* cfg generator

* update

* fix
2022-11-21 14:23:20 +08:00
Xinyu Wang 79a778689d
[Feature] Add SVT to dataset preparer (#1521)
* add svt to data preparer

* add svt parser test
2022-11-15 18:54:30 +08:00
Xinyu Wang 8864fa174b
[Feature] Add Dataset Preparer (#1484)
* add data preparer

* temporarily ignore data preparer test

* update

* fix comments

* update doc; add script to generate dataset zoo doc

* fix comments; update scripts

* apply comments

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

* apply comments

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

* coco parser

* fix comments

* add fileio tests

* fix test

* add tests for parsers and dumpers

* add test for data preparer

* fix a bug

* update icdar txt parser

* rename icdar txt parser

* fix comments

* fix test

* fix comments

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
Co-authored-by: liukuikun <641417025@qq.com>
2022-11-02 15:06:49 +08:00
Tong Gao e801df3471
[CI] Fix CI (#1365) 2022-09-05 16:58:31 +08:00
Tong Gao db6ce0d95e
[Refactor] ocr.py (#1344)
* [Feature] Add BaseInferencer]

* [Feature] Add Det&Rec Inferencer

* [Feature] Add KIEInferencer

* [Feature] Add MMOCRInferencer

* [Refactor] update ocr.py

* update links

* update two links

* remove ocr.py

* move ocr.py and add loadfromndarray

Co-authored-by: xinyu <wangxinyu2017@gmail.com>
Co-authored-by: liukuikun <liukuikun@sensetime.com>
2022-08-31 22:56:24 +08:00
Xinyu Wang 9bd5258513
[Refactor] Adapt to new dataflow (#1305)
* datasample->datasamples

* update rec data preprocessor

* rename datasamples

* update det preprocessor

* update metric

* update data_sample->data_samples in test

* update

* fix data preprocessor uts

* remove engine runner

* fix kie ut

* fix ut

* fix comments

* refactor evaluator ut

* apply comments

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

* remove useless

* apply comments

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

* apply comments

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

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-08-25 11:45:42 +08:00
Tong Gao e760dcd1dd
[Enhancement] Support loading different lmdb datasets in LoadImageFromLMDB (#1293)
* [Enhancement] Support loading different lmdb datasets in LoadImageFromLMDB

* add docstr
2022-08-22 16:43:25 +08:00
Tong Gao c7a4298c32
[Fix] Replace mmcv.fileio with mmengine (#1292) 2022-08-19 16:53:14 +08:00
Tong Gao 80d85c129f
[Fix] Rename tests (#1244) 2022-08-01 15:28:27 +08:00
gaotongxiao 05ff5d0489 Support pipeline assignment in ConcatDataset 2022-07-21 10:58:04 +08:00
wangxinyu 8c5e83c521 [Tests] rename tests 2022-07-21 10:58:04 +08:00
xinyu 254dbdd18a adapt to det package 2022-07-21 10:58:03 +08:00
jiangqing.vendor 993ee5a91c [TODO] Replace resize_cfg with resize_type 2022-07-21 10:57:31 +08:00
gaotongxiao f107991ac1 [TODO] Add LoadImageFromLMDB 2022-07-21 10:57:30 +08:00
gaotongxiao 914c8af7bf Revert "[TODO] Add LoadImageFromLMDB"
This reverts commit e716ae726f007f79effdf2d45b4955a887f3c1e3
2022-07-21 10:57:30 +08:00
jiangqing.vendor 19958fbf6f [TODO] Add LoadImageFromLMDB 2022-07-21 10:57:17 +08:00
jiangqing.vendor dae4c9ca8c Add MMDet2MMOCR MMOCR2MMdet 2022-07-21 10:57:17 +08:00
liukuikun d50d2a46eb [Processing]remove segocr and split processing 2022-07-21 10:57:17 +08:00
gaotongxiao 7490301877 Add KIE transforms 2022-07-21 10:55:45 +08:00
wangxinyu 52f0eefb2e [Transform] Add FixInvalidPolygon 2022-07-21 10:55:45 +08:00
gaotongxiao 9f2fabc35a Add WildReceipt Dataset 2022-07-21 10:55:45 +08:00
wangxinyu ab6e897c6b [Utils] Migrate datasets/utils 2022-07-21 10:55:45 +08:00
jiangqing.vendor 83ec5726d6 Add RecogDatasets 2022-07-21 10:55:44 +08:00
jiangqing.vendor 0dc4fda545 Add IcdarDatset 2022-07-21 10:55:44 +08:00
gaotongxiao 0fb0d7cb1a Fix UT 2022-07-21 10:51:03 +08:00
jiangqing.vendor 7be4dc1bca Refactor Parser 2022-07-21 10:51:03 +08:00
gaotongxiao 0716c97cf6 Add BoundedScaleAspectJitter 2022-07-21 10:51:03 +08:00
jiangqing.vendor 55c99dd0c1 [Update] Update TextDetRandomCropFlip 2022-07-21 10:51:02 +08:00
liukuikun d636adeb1f fix ut 2022-07-21 10:51:02 +08:00
gaotongxiao 8f7c0e2977 Fix imgaug 2022-07-21 10:51:02 +08:00
liukuikun f03ed3ce11 [Transform] RandomFlip 2022-07-21 10:51:01 +08:00
liukuikun dfe93dc7d2 [Transform] ScaleAspectJitter 2022-07-21 10:51:01 +08:00
gaotongxiao da175b44a4 [Fix] Fix RandomRotate 2022-07-21 10:51:01 +08:00
jiangqing.vendor 25e819f6bf [Fix] Fix TextDetRandomCrop 2022-07-21 10:51:01 +08:00
jiangqing.vendor b3b1ef146b [Fix] Check transform's unit test and visualize 2022-07-21 10:51:01 +08:00
gaotongxiao a4952a6dd6 Fix RandomCrop 2022-07-21 10:51:00 +08:00
liukuikun f47f3eff03 fix polygon type 2022-07-21 10:50:58 +08:00
jiangqing.vendor 4706cc7eca [Refactor] Refactor SquareResizePad 2022-07-21 10:50:58 +08:00
liukuikun 2f4679e908 LoadAnnotations 2022-07-21 10:50:58 +08:00
liukuikun 84a61ba816 RescaleToHeight and PadToWidth 2022-07-21 10:50:58 +08:00
liukuikun e8f57d6540 [Feature] PackDet/RecogInput 2022-07-21 10:50:58 +08:00