Commit Graph

325 Commits (main)

Author SHA1 Message Date
Tong Gao d0dc90253a
[Dataset Preparer] MJSynth (#1791)
* finialize

* finialize
2023-03-22 10:10:46 +08:00
Tong Gao 6d9582b6c7
[Docs] Fix quick run (#1775)
* Fix quick run

* fix ut

* debug

* fix

* fix ut
2023-03-22 10:09:54 +08:00
Tong Gao 9caacc76ee
[Enhancement] Deprecate file_client_args and use backend_args instead (#1765)
* tmp commit

* remove
2023-03-20 10:33:20 +08:00
Tong Gao e9b23c56ad
Cherry Pick #1205 (#1774)
* [Bug fix] box points ordering  (#1205)

* fix sort_points

* remove functools

* add test cases

* add descriptions for coordinates

* del

Co-authored-by: xinyu <wangxinyu2017@gmail.com>

* fix

---------

Co-authored-by: liferecords <yjmm10@yeah.net>
Co-authored-by: xinyu <wangxinyu2017@gmail.com>
2023-03-10 14:51:40 +08:00
Qing Jiang 75c06d34bb
[Dataset Preparer] Add SCUT-CTW1500 (#1677)
* update metafile and download

* update parser

* updata ctw1500 to new dataprepare  design

* add lexicon into ctw1500 textspotting

* fix

---------

Co-authored-by: liukuikun <641417025@qq.com>
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2023-03-08 17:32:00 +08:00
Tong Gao 33cbc9b92f
[Docs] Inferencer docs (#1744)
* [Enhancement] Support batch visualization & dumping in Inferencer

* fix empty det output

* Update mmocr/apis/inferencers/base_mmocr_inferencer.py

Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>

* [Docs] Inferencer docs

* fix

* Support weight_list

* add req

* improve md

* inferencers.md

* update

* add tab

* refine

* polish

* add cn docs

* js

* js

* js

* fix ch docs

* translate

* translate

* finish

* fix

* fix

* fix

* update

* standard inferencer

* update docs

* update docs

* update docs

* update docs

* update docs

* update docs

* en

* update

* update

* update

* update

* fix

* apply sugg

---------

Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2023-03-07 18:52:41 +08:00
Tong Gao 47f7fc06ed
[Feature] Support batch augmentation through BatchAugSampler (#1757)
* [Fix] RepeatAugSampler -> BatchAugSampler

* update docs
2023-03-07 11:29:53 +08:00
liukuikun 82f81ff67c
[Refactor] Refactor data converter and gather (#1707)
* Refactor dataprepare, abstract gather, packer

* update ic13 ic15 naf iiit5k cute80 funsd

* update dataset zoo config

* add ut

* finsh docstring

* fix coco

* fix comment
2023-03-03 15:27:19 +08:00
Tong Gao e9bf689f74
[Enhancement] Support batch visualization & dumping in Inferencer (#1722)
* [Enhancement] Support batch visualization & dumping in Inferencer

* fix empty det output

* Update mmocr/apis/inferencers/base_mmocr_inferencer.py

Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>

---------

Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2023-02-17 12:40:09 +08:00
liukuikun 1127240108
[Feature] Support auto import modules from registry. (#1731)
* [Feature] Support auto import modules from registry.

* limit mmdet version

* location parrent dir if it not exist
2023-02-17 10:28:34 +08:00
Tong Gao df0be646ea
[Enhancement] Speedup formatting by replacing np.transpose with torch.permute (#1719) 2023-02-16 14:14:03 +08:00
liukuikun f820470415
[Feature] Rec TTA (#1401)
* Support TTA for recognition

* updata readme

* updata abinet readme

* updata train_test doc for tta
2023-02-16 10:27:07 +08:00
Tong Gao 20a87d476c
[Fix] Fix some inferencer bugs (#1706)
* [Fix] Fix some inferencer bugs

* fix
2023-02-09 18:31:25 +08:00
Tong Gao 3b0a41518d
[Enhancement] Dynamic return type for rescale_polygons (#1702) 2023-02-02 19:18:33 +08:00
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 0aa5d7be6d
[Model] Add SVTR framework and configs (#1621)
* [Model] Add SVTR framework and configs

* update

* update transform names

* update base config

* fix cfg

* update cfgs

* fix

* update cfg

* update decoder

* fix encoder

* fix encoder

* fix

* update cfg

* update name
2023-01-06 16:07:06 +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
Yang Liu 5dbacfe202
[Feature] Add svtr encoder (#1483)
* add svtr backbone

* update backbone

* fix

* apply comments, move backbone to encoder

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2023-01-04 16:03:21 +08:00
Qing Jiang 65e746eb3d
[UT] Add missing unit tests (#1651)
* update

* remove code
2022-12-30 12:01:14 +08:00
Yang Liu 7e9f7756bc
[Feature] Add svtr decoder (#1448)
* add svtr decoder

* svtr decoder

* update

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-12-30 12:00:48 +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
Qing Jiang 419f98d8a4
[ASTER] Add ASTER decoder (#1625)
* add aster decoder

* aster decoder

* decoder

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-12-15 14:53:17 +08:00
Qing Jiang 0bd62d67c8
[ASTER] Add ASTER Encoder (#1239)
* [Docs] Limit extension versions (#1209)

* loss

* fix

* [update] limit extension versions

* add aster encoder

* aster encoder

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-12-15 14:53:05 +08:00
liukuikun ffe5237aa8
[Refactor] Refactor icdardataset metainfo to lowercase. (#1620) 2022-12-13 11:45:33 +08:00
liukuikun 3a0aa05d9c
[Feature] textspotting datasample (#1593)
* textspotting datasample

* rename
2022-12-06 14:03:32 +08:00
liukuikun 9ac9a227ec
[Improve] support head loss or postprocessor is None for only infer (#1594)
* support head loss or postprocessor is None for only infer

* default to None
2022-12-06 14:02:37 +08:00
liukuikun 5940d6bc9c
[Fix] textspotting ut (#1599) 2022-12-06 14:02:12 +08:00
Qing Jiang a12c215e85
[Refactor] Refactor TPS (#1240)
* [Docs] Limit extension versions (#1209)

* loss

* fix

* [update] limit extension versions

* refactor tps

* Update mmocr/models/textrecog/preprocessors/tps_preprocessor.py

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

* Update mmocr/models/textrecog/preprocessors/tps_preprocessor.py

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

* refine

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-12-05 14:51:40 +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
Tong Gao 37f3b88a05
[Feature] Add get_md5 (#1553)
* [Feature] Add get_md5

* fix
2022-11-18 15:22:47 +08:00
Tong Gao 34e97abcb0
[Enhancement] Polish bbox2poly (#1532) 2022-11-15 19:40:12 +08:00
Xinyu Wang 62ff782b71
[Fix] Fix ICDARTxtParser (#1529)
* fix

* fix comments

* fix

* fix

* fix

* delete useless
2022-11-15 19:18:14 +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
Wenqing Zhang f1dd437d8d
[Feature] support modified resnet structure used in oCLIP (#1458)
* support modified ResNet in CLIP and oCLIP

* update unit test for TestCLIPBottleneck; update docs

* Apply suggestions from code review

* fix

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-11-03 17:54:15 +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
liukuikun dfc17207ba
[Vis] visualizer refine (#1411)
* visualizer refine

* updata docs
2022-10-09 12:45:17 +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
Tong Gao 965f92f1e0
[Fix] Dataflow in mmdet wrapper (#1335)
* [Fix] Dataflow in mmdet wrapper

* fix typing

* fix ut

Co-authored-by: liukuikun <liukuikun@sensetime.com>
2022-08-31 19:03:59 +08:00
liukuikun bf042f8267
[Feture] kie visualizer (#1267)
* kie visualizer

* add  textspotting visualizer

* Fix

* fix

* Some fixes

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2022-08-29 20:13:06 +08:00
Tong Gao 9b368fe45c
[Fix] Rename VeryDeepVGG to MiniVGG (#1330)
* [Fix] Rename VeryDeepVGG to MiniVGG

* update
2022-08-25 16:35:59 +08:00
Tong Gao b32412a9e9
[Refactor] MMEngine directory tree (#1317) 2022-08-25 11:46:04 +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 1b5764b155
[Refactor] Rename base_xxx.py as base.py (#1322) 2022-08-25 11:20:42 +08:00
Tong Gao b81d58e70c
[Enhancement] Move dictionary (#1320) 2022-08-24 17:58:52 +08:00