a50b0c9fb9
[Feature] Support openset kie ( #498 )
...
* add openset kie dataset
* updare readme
* add anno convert script
* update docstring
* update script
* add & update docstring
* fix typo
* update docstring format
2021-11-11 14:47:38 +08:00
9f42d78db7
[Feature] Save filename in text detection test results ( #570 )
...
* save filename in textdet test results
* fix pytest
2021-11-11 14:07:07 +08:00
37f8fcd841
[Fix] Always map location to cpu when loading checkpoint ( #567 )
2021-11-10 14:03:03 +08:00
fdae159b89
[Fix] Fix py3.9 CI ( #563 )
2021-11-10 14:02:33 +08:00
1efdbda790
[Docs] Update docs menu ( #564 )
...
* update docs menu
* fix wrong links
2021-11-09 17:16:13 +08:00
5fb1268a06
unify task names in metafiles ( #548 )
2021-11-03 15:03:37 +08:00
dab5725d70
fix paper field in metafiles ( #550 )
2021-11-03 15:02:31 +08:00
80741e1479
[Feature] Add converter for the Open Images v5 text annotations by Krylov et al. ( #497 )
...
* Add converter for the OpenVINO annotations for Open Images by Krylov et al.
Open Images V5 Text Annotation and Yet Another Mask Text Spotter
Paper: https://arxiv.org/abs/2106.12326
* docs fix & add chinese docs
2021-10-28 16:49:36 +08:00
2f983eca83
[Fix] Fix PyTorch 1.6 incompatible checkpoints ( #540 )
...
* fix pt1.6 incompatible checkpoints
* fix satrn metafile
2021-10-27 19:12:52 +08:00
027cb7aac5
[Docs] TorchServe docs ( #539 )
...
* Add torchserve docs
* polish contents and fix the visiualization bug in torchserve testing
* polish content
2021-10-22 16:49:28 +08:00
5671939bbc
[Fix] fix converting to onnx bug because of changing key from img_shape to resize_shape ( #523 )
...
* fix pytorch to onnx for mmocr0.3.0
* fix pytorch to onnx for mmocr0.3.0
* modify deployment doc and fix scale_factor bug and test psenet and panet
* fix trt part
* init detector with **(cfg.model)
Co-authored-by: AllentDan <AllentDan@yeah.net>
2021-10-20 19:55:57 +08:00
75d32504e0
[Feature] Add codespell pre-commit hook and fix typos ( #520 )
...
* add spellcheck hook
* fix typos
2021-10-13 22:56:11 +08:00
7d36a7e882
[Feature] Add torchserve support for text detection and recognition ( #522 )
...
* add torchserve
* add test torchserve
* delete unused postprocess and delete parse_results
* fix bug
* fix bug
* fix bug
* Update tools/deployment/test_torchserve.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update tools/deployment/mmocr_handler.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-10-13 22:21:16 +08:00
abfe75dc62
[Fix] Update checkpoint links for SATRN ( #518 )
2021-10-10 11:26:37 +08:00
66ce0efee0
[Docs] Add Chinese Translation of deployment.md. ( #506 )
...
* Add Chinese Translation of deployment.md.
* final fix
Co-authored-by: zhaoyuxuan <zhaoyuxuan@zhaoyuxuandeMacBook-Pro.local>
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-09-28 17:11:15 +08:00
06c726c130
[Enhancement] Use container in CI ( #502 )
...
* use container
2021-09-28 14:05:10 +08:00
f72012092f
[Docs] Enhancement & sync Chinese docs ( #492 )
...
* reorder navigation links, add links to twitter and zhihu
* update chinese docs
2021-09-28 11:32:44 +08:00
aa4b5b5574
[Fix] Allow invalid polygons in intersection and union by default ( #471 )
...
* Allow invalid polygons in intersection and union by default
* update docs
2021-09-28 10:57:12 +08:00
8c72d80164
[Fix] Fix image export in test.py for KIE models ( #486 )
...
* Fix image export in test.py for sdmgr model
* fix pretrained
2021-09-18 18:24:55 +08:00
38bdc10f22
[Docs] Chinese docs for install.md ( #364 )
...
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-09-14 11:47:38 +08:00
e9a5350f47
[Enhancement] Update model-index.yml ( #484 )
2021-09-08 12:51:44 +08:00
de83b271eb
add py1.9 CI ( #475 )
2021-09-08 11:45:51 +08:00
e23b765956
[Docs] Improve docs style ( #474 )
...
* new theme
* add theme
* update zh_cn
* improve docs style
* use customized
* fix
* update req
* docs
* Update docs
* update conf
* update
* update layout
* disable logo url
* free version limit
* update conf
* Fix api ref
* fix version
2021-09-08 11:40:51 +08:00
76c9570b39
[Enhancement] add CITATION.cff ( #476 )
2021-09-01 20:01:30 +08:00
4c43f3fc47
[Docs] Fix a model link and add the metafile for SATRN ( #473 )
...
* update model link
* add metafile
2021-09-01 03:01:11 +00:00
f59ac27022
[Enhancement] Use bounding box around polygon instead of within polygon ( #469 )
...
* use bounding box around poygon instead of within polygon
* fix yapf error
* fix: re-enable yapf
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-08-31 13:05:40 +08:00
b2ae04614c
[Fix] Fix cuda CI ( #472 )
2021-08-31 13:00:59 +08:00
cc0d4d32fd
[Feature] Support Chinese for kie show result ( #464 )
...
* support Chinese for kie show result
* update visualize
* add arg for draw_texts
* update font_size
2021-08-30 16:45:43 +08:00
f9d158fafa
Bump version to 0.3.0 ( #461 )
...
* Bump version to 0.3.0
* update changelog
2021-08-25 16:43:31 +08:00
c0728c49b8
[Docs] Refactor docs ( #409 )
2021-08-25 16:41:07 +08:00
0881c2d2a2
[Fix] Add zero division handler in poly utils, remove Polygon3 ( #448 )
...
* Add check to avoid zero div in iou computation
* replace polygon3 with shapely
* remove req of Polygon3
2021-08-25 13:14:58 +08:00
7c1bf45c63
[Fix] fix inconsisent training/eval state after SyncBN->BN ( #453 )
2021-08-25 13:14:03 +08:00
961fbb6ca5
[Feature] Support persistent_workers option in data loader ( #459 )
...
* Support persistent_workers option in data loader
Signed-off-by: lizz <lizz@sensetime.com>
* Update test.py
* yapf bug
Signed-off-by: lizz <lizz@sensetime.com>
2021-08-25 13:12:53 +08:00
ef111b7c55
[Enhancement] Update demo/ ( #454 )
...
* update demo
* rm demo/readme.md
2021-08-25 03:45:47 +00:00
4e6f70d7c4
[Enhancement] Replace lanms-proper with lanms-neo ( #455 )
2021-08-23 15:32:33 +08:00
d683b14283
[Fix] Totaltext_converter: skip invalid annotations ( #438 )
...
* [Fix] Skip invalid annoataions
2021-08-20 11:23:05 +08:00
8f377f2fa2
[Feature] Add Satrn ( #405 )
...
* Add SATRN
* Create satrn_small_academic.py
* Update README.md
* change config name
* Update mmocr/models/textrecog/backbones/shallow_cnn.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update configs/textrecog/satrn/satrn_academic.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update configs/textrecog/satrn/satrn_small.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update shallow_cnn.py
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update test_ocr_encoder.py
* change keep_aspect_ratio=False
* Update transformer_layer.py
* Update configs/textrecog/satrn/satrn_small.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update configs/textrecog/satrn/satrn_academic.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update mmocr/models/textrecog/layers/transformer_layer.py
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update transformer_layer.py
* Apply suggestions from code review
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
* Update transformer_layer.py
* update satrn readme
* add satrn to ocr.py
* add satrn_sm and fix configs
* add a test for config
* add copyright info
* use mmocr registry
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2021-08-19 22:02:58 +08:00
bc4c781e8f
[fix] fix when dbnet polygon points number less than 4 there is a uclip bug ( #443 )
2021-08-19 12:05:28 +00:00
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
b53098dc5c
update citation information ( #440 )
2021-08-19 09:05:30 +00:00
b8f7ead74c
[Enhancement] Add copyright info ( #439 )
...
* add copyright info
2021-08-17 17:39:30 +08:00
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
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
906faec372
update digit_version and add a test ( #427 )
2021-08-12 08:53:45 +00:00
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
532e8f808d
Fix #416 : Fix undesired assignment to "pretrained" in test.py ( #418 )
...
* fix ocr linting
* fix undesired assignment to pretrained key
2021-08-10 14:07:41 +00:00
f62b4513f5
[Fix] Fix a wrong link in ocr.py ( #417 )
2021-08-09 19:52:30 +08:00
304b48e187
[Docs] Full screen support in readthedocs ( #413 )
...
* add full screen css
* New dir structure
2021-08-09 11:16:43 +00:00
8227fec826
[Fix] Fix single model inference in ocr.py ( #411 )
2021-08-04 16:58:47 +08:00
2cbb1a6cd6
[Docs] Fix docs ( #395 )
...
1. Supplement links for ct80 and svtp
2. Simplify getting_started.md
3. clarify batch_mode support in demo.md
2021-08-04 14:26:16 +08:00
Hongbin Sun
Tong Gao
Darwin Bautista
liukuikun
liukuikun
fatfishZhao
EricTian
Alexander Soare
lizz
LvTianlei
yuexy
Alireza Kashani