Commit Graph

55 Commits (a09437adaa13a067c664928160222c46850534d3)

Author SHA1 Message Date
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
GT 9582f93c70
Support ocr box stitching (#290)
* ocr box stitching

* rename a varible

* add a demo

* move functions to box_util
add test
2021-06-19 19:28:14 +08:00
Jianyong Chen 3bfbb2b619
fix #113: textsnake decode error (#264)
* fix #113: textsnake decode error

* fix unittest test cases
2021-06-15 08:09:05 +00:00
lizz b10b6408ef
Add list_from_file and list_to_file (#226)
* Add list_from_file and list_to_file

Signed-off-by: lizz <lizz@sensetime.com>

* Add test list_to_file and list_from_file

* more

* Fix tests
2021-05-24 06:01:42 +00:00
Zyq-scut 8c60e4b4d1
update fcenet (#214)
* update fcenet

* fix typos

* fix
2021-05-18 21:52:34 +08:00