Commit Graph

325 Commits (main)

Author SHA1 Message Date
Tong Gao c093c687a7
[Config] Refactor base config (part 3) (#1316)
* [Config] Refactor KIE

* fix ut

* fix edge classes
2022-08-24 17:30:05 +08:00
Tong Gao 1860a3a3b6
[Feature] Use BCEWithLogitsLoss as many as possible for AMP (#1309)
* Use BCEWithLogitsLoss as many as possible for AMP

* fix

* Optimize DBNet

* fix docstr

* Use branch in dbhead, fix missing data_samples in textdethead

* fix
2022-08-23 19:17:27 +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
liukuikun d27b2fd84f
[Config] rec config refine (#1306)
* refine rec config

* fix mj-sub
2022-08-22 16:42:56 +08:00
Tong Gao 240bf06ddd
[Fix] unit tests due to new config names (#1303) 2022-08-22 15:02:52 +08:00
Tong Gao c7a4298c32
[Fix] Replace mmcv.fileio with mmengine (#1292) 2022-08-19 16:53:14 +08:00
Tong Gao 0d9b40706c
[Fix] Remove dependency on MMCV registry (#1261)
* [Fix] Remove dependency on MMCV registry

* fix
2022-08-19 11:19:07 +08:00
liukuikun 587566d2c2
[Fix] evaluator ut (#1285) 2022-08-16 17:34:07 +08:00
liukuikun bcc245efd3
[Evaluator] MultiDatasetEvaluator (#1250)
* multi datasets evalutor

* fix comment

* fix typo
2022-08-16 16:48:04 +08:00
Tong Gao 7b25b62c21
[Fix] Fix dictionary docstr and remove unncessary kwargs (#1276)
* [Fix] Fix dictionary docstr and remove unncessary kwargs

* fix

* fix
2022-08-11 11:14:17 +08:00
liukuikun ef683206ed
[Api] vis hook and data flow api (#1185)
* vis hook and data flow api

* fix comment

* add TODO for merging and rewriting after MultiDatasetWrapper
2022-08-08 11:37:46 +08:00
Tong Gao 80d85c129f
[Fix] Rename tests (#1244) 2022-08-01 15:28:27 +08:00
Qing Jiang 48cc575507
[Rename] Rename mmocr/data to mmocr/structures (#1234)
* rename mmocr/data to mmocr/structures

* update lint
2022-08-01 10:59:53 +08:00
Mountchicken 717460055c Revert "[update] Rename tests"
This reverts commit 0183a73b75.
2022-07-27 17:52:35 +08:00
liukuikun 2cca103b93
[Fix] fix load error (#1212) 2022-07-26 16:53:11 +08:00
liukuikun bc043101fe
[Fix] load error (#1207) 2022-07-26 10:48:10 +08:00
liukuikun 83e4fb10ee
[Fix] fix empty image and resize small image (#1200) 2022-07-25 19:21:09 +08:00
Tong Gao 6f30020eec
[CI] Add CI (#1176)
* [CI] Add CI

* update init

* fix lint

* fix lint

* fix linting

* fix linting

* fix linting

* fix

* fix

* fix

* fix

* fix

* fix

* disable github ci

* fix

* Update .circleci/test.yml

Co-authored-by: Qing Jiang <mountchicken@outlook.com>

* fix

* fix

Co-authored-by: Qing Jiang <mountchicken@outlook.com>
2022-07-21 14:28:57 +08:00
jiangqing.vendor e303404215 [Fix] Fix some config errors 2022-07-21 10:58:04 +08:00
liukuikun f5e93d0eba fix mmdet rename data_elements to structures 2022-07-21 10:58:04 +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
gaotongxiao 1a167ff317 Migrate tests 2022-07-21 10:58:03 +08:00
gaotongxiao 3980ead987 Rename & Migrate tests for EncoderDecoderRecognizer 2022-07-21 10:58:03 +08:00
jiangqing.vendor 3734527d38 [TODO] Add char2idx 2022-07-21 10:57:31 +08:00
jiangqing.vendor dc84187311 [TODO] Replace loss_module with module_loss 2022-07-21 10:57:31 +08:00
gaotongxiao 1cbc42eceb mmocr.model.textdet.losses.common -> mmocr.model.common.losses 2022-07-21 10:57:31 +08:00
gaotongxiao 8bce19218e Migrate some tests of utils 2022-07-21 10:57:31 +08:00
jiangqing.vendor 993ee5a91c [TODO] Replace resize_cfg with resize_type 2022-07-21 10:57:31 +08:00
gaotongxiao eb2d5b525a Migrate part of old_tests 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
gaotongxiao bf517b63e8 Update outputs of DBHead and split_results in BaseTextDetPostprocessor 2022-07-21 10:57:17 +08:00
gaotongxiao 41a642bc7b Refactor testing 2022-07-21 10:57:17 +08:00
jiangqing.vendor ae4ba012a8 [MMDet] DetWrapper 2022-07-21 10:57:17 +08:00
jiangqing.vendor dae4c9ca8c Add MMDet2MMOCR MMOCR2MMdet 2022-07-21 10:57:17 +08:00
wangxinyu de616ffa02 [TODO] Add tests for box center/diag dist; recover some evaluation utils 2022-07-21 10:57:17 +08:00
jiangqing.vendor 67e4085915 [Fix] Fix crop_polygon 2022-07-21 10:57:17 +08:00
wangxinyu 1d1f664e9a Update max_seq_len 2022-07-21 10:57:17 +08:00
liukuikun 25faa7d1f1 [Content] refactor vis content 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 a844b497db [ABINet] Refactor ABILoss 2022-07-21 10:57:17 +08:00
xinyu 23e1f2432a update utils 2022-07-21 10:57:16 +08:00
liukuikun 966e2ca9de [Content] data_structure content 2022-07-21 10:55:47 +08:00
liukuikun 5381b1d105 refactor loop content 2022-07-21 10:55:46 +08:00
gaotongxiao b8d472b77b Allow invalid ignore chars in BasePostprocessor 2022-07-21 10:55:46 +08:00
liukuikun 4e603f0531 [Remove] remove useless eval and refactor metrics content 2022-07-21 10:55:46 +08:00
gaotongxiao 02c6802312 ABI Decoders 2022-07-21 10:55:46 +08:00
gaotongxiao 2cb55550cd update encoders 2022-07-21 10:55:46 +08:00
gaotongxiao 2df8cb89a4 [SDMGR] Add SDMGR framework 2022-07-21 10:55:46 +08:00
xinyu 2fe534b178 add docstring and test for fill hole 2022-07-21 10:55:46 +08:00
liukuikun b828d654a9 [Encoder] scanner encoder 2022-07-21 10:55:46 +08:00
gaotongxiao 460f068891 [DRRG] DRRG framework 2022-07-21 10:55:46 +08:00
gaotongxiao 9470821aa0 [DRRG] DRRG head 2022-07-21 10:55:46 +08:00
gaotongxiao e0992a7fae [DRRG] DRRG loss 2022-07-21 10:55:46 +08:00
gaotongxiao ed9e8d150c [DRRG] DRRG postprocessor 2022-07-21 10:55:46 +08:00
gaotongxiao 9db0941837 [SDMGR] Add SDMGR Head 2022-07-21 10:55:46 +08:00
gaotongxiao eaf7f6bf0c [SDMGR] postprocessor 2022-07-21 10:55:46 +08:00
gaotongxiao e23a2ef089 Add SDMGR Loss 2022-07-21 10:55:46 +08:00
liukuikun 622e65926e [Decoder]Robust scanner decoder 2022-07-21 10:55:45 +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 c5364f843d Add F-metric 2022-07-21 10:55:45 +08:00
xinyu 02a43d234e clean point utils todos 2022-07-21 10:55:45 +08:00
liukuikun b20bcc47b3 multi loop 2022-07-21 10:55:45 +08:00
xinyu 6ff567bb08 fix np warnings & fix nose_parameterized warnings 2022-07-21 10:55:45 +08:00
gaotongxiao 9f2fabc35a Add WildReceipt Dataset 2022-07-21 10:55:45 +08:00
gaotongxiao 09169f32ee Support skipping unknown tokens in dictionary 2022-07-21 10:55:45 +08:00
wangxinyu 4b185d3347 [Model] Add MobilenetV2 Backbone 2022-07-21 10:55:45 +08:00
wangxinyu ab6e897c6b [Utils] Migrate datasets/utils 2022-07-21 10:55:45 +08:00
wangxinyu ef98df8052 [Utils] Migrate core/evaluation/utils.py 2022-07-21 10:55:45 +08:00
jiangqing.vendor 83ec5726d6 Add RecogDatasets 2022-07-21 10:55:44 +08:00
liukuikun b955df9904 [Detector] refactor basedetector 2022-07-21 10:55:44 +08:00
jiangqing.vendor 0dc4fda545 Add IcdarDatset 2022-07-21 10:55:44 +08:00
liukuikun 988fea441b [Recognizer] refactor baserecognizer 2022-07-21 10:55:43 +08:00
gaotongxiao 0fb0d7cb1a Fix UT 2022-07-21 10:51:03 +08:00
liukuikun 41d9c741cd [Decoder] sar decoder 2022-07-21 10:51:03 +08:00
liukuikun 47771788f0 [Encoder] sar encoder 2022-07-21 10:51:03 +08:00
jiangqing.vendor 7be4dc1bca Refactor Parser 2022-07-21 10:51:03 +08:00
jiangqing.vendor 0bf1ce88c2 [FCENet] Add FCE Postprocessor 2022-07-21 10:51:03 +08:00
gaotongxiao cd4e520cb9 Add KIEDataSample 2022-07-21 10:51:03 +08:00
wangxinyu c0c0f4b565 [PSE] PSE Postprocessor 2022-07-21 10:51:03 +08:00
wangxinyu 4a04982806 [PSE] PSE Loss 2022-07-21 10:51:03 +08:00
gaotongxiao 0716c97cf6 Add BoundedScaleAspectJitter 2022-07-21 10:51:03 +08:00
wangxinyu 00ba46b5b9 [PSE] PSE Neck FPNF 2022-07-21 10:51:03 +08:00
wangxinyu 05990c58d9 [Refactor] PSE Head 2022-07-21 10:51:03 +08:00
gaotongxiao 490d6cd806 Fix MaskedCELoss 2022-07-21 10:51:03 +08:00
jiangqing.vendor fded755af2 [FCENet] Add FCENet loss 2022-07-21 10:51:03 +08:00
Mountchicken 17606c25fc add fce head 2022-07-21 10:51:02 +08:00
liukuikun 200899b2a0 [PAN Postprocessor]pan postprocessor 2022-07-21 10:51:02 +08:00
jiangqing.vendor 13920924ce [MASTER] Add master plugin 2022-07-21 10:51:02 +08:00
liukuikun b6e031666b [PAN] pan head 2022-07-21 10:51:02 +08:00
jiangqing.vendor 55c99dd0c1 [Update] Update TextDetRandomCropFlip 2022-07-21 10:51:02 +08:00
liukuikun 7c3789d64e [PANLoss] pan loss 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
gaotongxiao 21d0dd71dc [TextSnake] Refactor textsnake loss 2022-07-21 10:51:02 +08:00
wangxinyu bf7c738798 [TextSnake] TextSnake Neck 2022-07-21 10:51:02 +08:00