Commit Graph

1070 Commits (e9bf689f74dd9929708aa266763fa1e691a90a0a)
 

Author SHA1 Message Date
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
liukuikun 7cea6a6419
[Enchancement] Only keep meta and state_dict when publish model (#1729)
* Only keep meta and state_dict when publish model

* simpy code
2023-02-15 19:45:12 +08:00
vansin 3240bace4a
docs: fix the head show in readme (#1727) 2023-02-15 16:07:24 +08:00
vansin b21d2b964a
docs: Add twitter discord medium youtube link (#1724) 2023-02-15 11:10:41 +08:00
Qing Jiang 332089ca11
[Fix] Add missing softmax in ASTER forward_test (#1718)
* add missing softmax

* update
2023-02-13 10:32:55 +08:00
Ikko Eltociear Ashimine b3be8cfbb3 Fix typo in ilst_converter.py (#1711)
* Fix typo in ilst_converter.py

splited -> splitted

* Fix typo in ilst_converter.py
2023-02-11 09:09:38 +08:00
Xinyu Wang d25e061b03
[Fix] textocr ignore flag (#1712) 2023-02-10 16:09:00 +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 d8e615921d
[Fix] Detect intersection before using shapley.intersection to eliminate spurious warnings (#1710)
* [Fix] Detect intersection before using shapley.intersection to eliminate spurious warnings

* Update polygon_utils.py
2023-02-08 10:36:11 +08:00
Tong Gao 2a2cab3c8c
[Checkpoints] Add ST-pretrained DB-series models and logs (#1635)
* [Fix] Auto scale lr

* update
2023-02-06 15:16:08 +08:00
liukuikun c870046a4a
[Fix] change cudnn benchmark to false (#1705) 2023-02-03 18:57:12 +08:00
Kevin Wang edf085c010
[Feature] TextRecogCropConverter add crop with opencv warpPersepective function (#1667)
* [Feature] TextRecogCropConverter add crop with opencv warpPersepective function.

* [Fix] fix some pr problems

* Apply suggestions from code review

---------

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2023-02-03 17:04:37 +08:00
Tong Gao c3aef21eea
[Enhancement] Revise upstream version limit (#1703)
* [Enhancement] Revise upstream version limit

* update
2023-02-03 16:39:32 +08:00
Tong Gao 03a23ca4db
[Docs] Remove unsupported datasets in docs (#1670) 2023-02-02 19:47:10 +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 2d743cfa19
[Model] SPTS (#1696)
* [Feature] Add RepeatAugSampler

* initial commit

* spts inference done

* merge repeat_aug (bug in multi-node?)

* fix inference

* train done

* rm readme

* Revert "merge repeat_aug (bug in multi-node?)"

This reverts commit 393506a97c.

* Revert "[Feature] Add RepeatAugSampler"

This reverts commit 2089b02b48.

* remove utils

* readme & conversion script

* update readme

* fix

* optimize

* rename cfg & del compose

* fix

* fix
2023-02-01 11:58:03 +08:00
Xinyu Wang 2b5cdbdbfc
update owners (#1699) 2023-02-01 10:07:17 +08:00
Tong Gao a82fc66812
[Feature] Add RepeatAugSampler (#1678)
* [Feature] Add RepeatAugSampler

* fix
2023-01-31 19:42:48 +08:00
Kevin Wang bed778fc3f
[Fix] fix isort pre-commit error (#1697)
* [Fix] fix isort pre-commit error

* [Fix] add mmengine to setup.cfg and rerun pre-commit
2023-01-31 18:50:13 +08:00
liukuikun 689ecf0f5f
fix LoadOCRAnnotation ut (#1695) 2023-01-31 11:23:49 +08:00
Tong Gao bf41194965
[Docs] Add notice for default branch switching (#1693) 2023-01-30 10:27:02 +08:00
tripleMu dff97edaad
fix lint (#1694)
fix lint
2023-01-29 17:42:34 +08:00
Qing Jiang 50f55c2976
[Fix] Fix a minor error in docstring (#1685) 2023-01-28 11:11:54 +08:00
Tong Gao b3f21dd95d
[Fix] Explictly create np object array for compatability (#1691) 2023-01-28 11:08:01 +08:00
liukuikun 7f4a1eecdc
abcnetv2 inference (#1657)
* abcnetv2 inference

* update readme
2023-01-18 18:37:19 +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 b64565c10f
[Fix] Update dockerfile (#1671)
* [Fix] Update dockerfile

* [Fix] Update dockerfile
2023-01-16 15:34:38 +08:00
AllentDan 39f99ac720
[Doc] update the link of DBNet (#1672)
* update the link of DBNet_r50

* update the link of DBNet_r50-oclip
2023-01-12 11:06:53 +08:00
Tong Gao 27b6a68586
Bump version to 1.0.0rc5 (#1662)
* Bump version to 1.0.0rc5

* fix

* update
2023-01-06 17:35:07 +08:00
Tong Gao 37dca0600a
Issue Template (#1663)
* [Template] Refactor issue template (#1449)

* Refactor issue template

* add contact

* [Template] issue template (#1489)

* improve issue template

* fix comment

Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2023-01-06 17:29:28 +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
Tong Gao d679691a02
[CI] Remove support for py3.6 (#1660)
* [CI] Remove support for py3.6

* update

* fix
2023-01-06 14:22:38 +08:00
liukuikun acae8da223
[Docs] updata abcnet doc (#1658)
* updata abcnet doc

* updata link

* updata link

* updata config name

* add link for data
2023-01-06 10:31:08 +08:00
liukuikun 4d5ed98177
[Feature] ABCNet train (#1610)
* abcnet train

* fix comment

* updata link

* fix lint

* fix name
2023-01-05 18:53:48 +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
Tong Gao 53e72e4440
[Metafile] Add Aliases to models (#1611)
* [Metafile] Add Aliases to models

* update
2022-12-29 17:44:32 +08:00
Ferry Huang 1413b5043a
[Feature] CodeCamp #116 Add SROIE to dataset preparer (#1639)
* added sroie/metafile.yml

* add sample_anno.md and textdet.py

* modify and add all

* fix lint

* fix lint

* fix lint

* Update mmocr/datasets/preparers/data_converpyter.

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

* fix the reviewed

* add comment of try to sroie_parser.py

* modify data_obtainer.py

* fix lint errors

* fix download link

Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-12-29 16:52:51 +08:00
Qing Jiang b79382cd6b
[Feature] CodeCamp #115 Add NAF to dataset preparer (#1609)
* add naf converter

* fix test

* update

* use fuzzy search instead

* update

* update
2022-12-29 15:19:49 +08:00
Xinyu Wang e3fd570687
[Docs] add translation (#1647) 2022-12-29 09:52:55 +08:00
liukuikun 9baf440d7a
[Feature] ConditionApply (#1646) 2022-12-28 11:53:32 +08:00
Tong Gao 89606a1cf1
[Configs] Totaltext cfgs for DB and FCE (#1633)
* fcenet configs

* dbnet config

* update fcenet config

* update dbnet config

* Add readme and metafile
2022-12-28 11:51:38 +08:00
Tong Gao e1aa1f6f42
[Fix] CI (#1636) 2022-12-28 11:51:06 +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