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
7b16754006
[Docs] Fix API reference & version, support exporting pdf & epub ( #406 )
...
* add mmdet
* remove uncessary autodoc mock import
* update docs version
* bind docs version to version.py
* support pdf and epub
* fix readthedocs
2021-08-04 14:24:52 +08:00
b812b7a57c
[Feature] Add PyTorch 1.9 tests in CI ( #401 )
2021-08-04 14:23:53 +08:00
72a5c31577
add mim support ( #394 )
2021-08-04 14:21:41 +08:00
68ec3f5519
Add kie image demo and docs. ( #374 )
...
* Add kie_image_demo.py
* Add kie demo docs
* Add brief instructions for kie image demo
* Add ann file field and return data for kie image demo
* Follow lint and import rules
* Fix bugs, reuse functions in KIEDataset, and use a new demo pic
* Add config-dir and fix indexing bug in ocr script
* [Feature] Improve ocr.py
1. Add box stitching back to ocr.py
2. Add config_dir which allows users to specify the default config path
3. Warn users when overriding parameters are set
4. Allow users to use customized checkpoint files
* Add docs for new ocr.py
* Add docs for merge
* Support kie in ocr.py
* Merged kie to ocr.py
* update docs, remove unsupported unvisual sdmgr
* Update mmocr/apis/inference.py
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
* Apply suggestions from code review
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
* fix linting
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-08-04 14:20:13 +08:00
f24be6c614
fix bug of dice loss: loss always 1 if empty target ( #408 )
2021-08-04 06:05:31 +00:00
3707d67106
support batch inference for crnn and segocr ( #407 )
...
* support batch inference for crnn and segocr
2021-08-03 16:52:58 +08:00
9c76f7b5cc
[Feature] Improve ocr.py ( #400 )
...
* [Feature] Improve ocr.py
1. Add box stitching back to ocr.py
2. Add config_dir which allows users to specify the default config path
3. Warn users when overriding parameters are set
4. Allow users to use customized checkpoint files
5. Add docs
2021-08-03 15:57:30 +08:00
a2af2b4d31
Add output option to det_recog_show_result method ( #386 )
...
* add out_file arg
* fix small bug
* Update visualize.py
2021-07-25 12:29:40 +08:00
76785e185a
User friendly API v2 + Docs ! ( #371 )
...
* major update
- Refactor code
- Support for folder and list/tuple or np.arrays or img paths
- Better export method
* feature update
- Batch size support
- More refactoring
* added docs
* Optimize docs structure, fix improper layout in readthedocs
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-07-25 00:22:27 +08:00
200dfe5fe2
[Fix]: Fix dim mismatch in batch test/inference of DBNet ( #383 )
...
Modification
1.change the dbnet output dim
2.change the input dim for DBloss
2021-07-24 09:14:03 +08:00
02d657d141
[Fix] upgrade version requirement of mmdet to 2.14.0 to avoid known bugs ( #382 )
2021-07-21 18:01:55 +08:00
969c65165f
[Fix] Remove depreciated pretrained in kie_test_imgs ( #381 )
2021-07-21 15:18:59 +08:00
c1ae3a41e5
Bump version to 0.2.1 ( #378 )
2021-07-20 23:19:07 +08:00
Tong Gao
fatfishZhao
EricTian
Alexander Soare
Hongbin Sun
lizz
LvTianlei
liukuikun
yuexy
Alireza Kashani
Omkar Manjrekar
Jiaqi Duan
SamAyala