59 Commits

Author SHA1 Message Date
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
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
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
wangxinyu
8c5e83c521 [Tests] rename tests 2022-07-21 10:58:04 +08:00
gaotongxiao
8bce19218e Migrate some tests of utils 2022-07-21 10:57:31 +08:00
gaotongxiao
eb2d5b525a Migrate part of old_tests 2022-07-21 10:57:31 +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
xinyu
23e1f2432a update utils 2022-07-21 10:57:16 +08:00
xinyu
2fe534b178 add docstring and test for fill hole 2022-07-21 10:55:46 +08:00
wangxinyu
52f0eefb2e [Transform] Add FixInvalidPolygon 2022-07-21 10:55:45 +08:00
xinyu
02a43d234e clean point utils todos 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
b3b1ef146b [Fix] Check transform's unit test and visualize 2022-07-21 10:51:01 +08:00
gaotongxiao
3a9f9e6b61 Handling a corner case in offset_polygon 2022-07-21 10:51:00 +08:00
gaotongxiao
cd3d173b18 [DBNet] Add DBLoss 2022-07-21 10:51:00 +08:00
gaotongxiao
0f0f68baf1 Fix crop_polygon, Resize and poly2shapely 2022-07-21 10:50:59 +08:00
liukuikun
fe43259a05 [Refactor] train and test 2022-07-21 10:50:59 +08:00
gaotongxiao
7e7a526f37 New Hmean-iou metric 2022-07-21 10:50:59 +08:00
liukuikun
2f4679e908 LoadAnnotations 2022-07-21 10:50:58 +08:00
wangxinyu
178030bad6 [Refactor] Refactor transform.RandomRotate 2022-07-21 10:50:57 +08:00
jiangqing.vendor
f29853d9cd [Feature] Add Resize 2022-07-21 10:50:57 +08:00
gaotongxiao
6478499073 add mode to rescale_polygon(s) 2022-07-21 10:50:57 +08:00
gaotongxiao
df2f7b69db Add recognition data migrator 2022-07-21 10:50:57 +08:00
gaotongxiao
cb85f857aa Add BaseTextDetPostProcessor 2022-07-21 10:50:57 +08:00
gaotongxiao
f0c6d44ce8 Add dump_ocr_data 2022-07-21 10:50:56 +08:00
gaotongxiao
b5fc589320 Add CI 2022-07-21 10:50:55 +08:00
gaotongxiao
69e6c80558 move tests to old_tests, add empty test folders 2022-07-21 10:50:55 +08:00
garvan2021
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
JiangQing
680dff373e
[Feature] Support jsonl in recognition converter (#844) 2022-03-18 09:22:32 +08:00
Tong Gao
fb77352eb2
[Fix] Standardize the type of torch.device in ocr.py (#800) 2022-03-03 14:18:33 +08:00
Tong Gao
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
Tong Gao
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
liukuikun
d871eed79e
[Enhancement] Add setup multi-processing both in train and test. (#757) 2022-01-26 20:34:19 +08:00
yuexy
82488024db [Feature] Add bezier_to_polygon and sort_points to box_util (#703) 2022-01-17 15:26:55 +08:00
Hongbin Sun
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
Hongbin Sun
c6c4acd39b
[Fix] fix type check (#650)
* fix check type

* add notice
2021-12-09 12:45:59 +08:00
Tong Gao
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
Tong Gao
7c1bf45c63
[Fix] fix inconsisent training/eval state after SyncBN->BN (#453) 2021-08-25 13:14:03 +08:00
Tong Gao
7571763376
[Refactor] Use MMOCR's registry (#436)
* [Refactor] Use MMOCR's registry

1. Define MMOCR's registries as a child of MMDet's
2. Register all models to MMOCR's own registries
3. Modify some model configs so that some models in MMDet can be
   correctly located
4. Remove some outdated demo scripts

* add detectors
2021-08-19 19:17:15 +08:00
Tong Gao
b8f7ead74c
[Enhancement] Add copyright info (#439)
* add copyright info
2021-08-17 17:39:30 +08:00
liukuikun
9b5b25ef71
[Fix] fix a bug of DBNet when text repr type is poly (#421)
* fix dbnet bug when text repr type is poly

* add db_decode unit test
2021-08-13 21:39:53 +08:00
Tong Gao
80a0536c7c
[Enhancement] Add tests for ocr.py (#428)
* minor fix to ocr.py

* add test for ocr.py
2021-08-13 09:55:33 +08:00
Tong Gao
906faec372
update digit_version and add a test (#427) 2021-08-12 08:53:45 +00:00
Tong Gao
7bbb14f0d1
[Fix] Replace SyncBN with BN for inference (#420)
* add revert_sync_batchnorm

* replace SyncBN in inference and test scripts

* add tests

* hide BatchNormXd
2021-08-10 22:19:17 +08:00
Hongbin Sun
44fed8590b
fix bug: remove tail whitespace from gt label (#358)
* fix bug: remove tail whitespace from gt label

* remove useless

* refactor
2021-07-09 13:24:34 +00:00