Commit Graph

181 Commits (3b9306ca7964ba3f297f01802985b862fc3ef191)
 

Author SHA1 Message Date
Hongbin Sun 3b9306ca79
Hbsun/demo doc (#168)
* move demo.md to demo/docs

* update link in md
2021-05-12 09:57:53 +08:00
Hongbin Sun 45d22b12c6
update demo docs (#166)
* update demo docs

* update docs
2021-05-11 21:13:19 +08:00
quincylin1 3ccec09106
add mmocr demo documentation (#151)
* add mmocr demo documentation

* Update text_det_demo.md

* Update text_recog_demo.md

* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-11 16:38:57 +08:00
lizz 6a1d1020f6
Add a message when annotation is empty (#165)
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-11 16:38:30 +08:00
Hongbin Sun 85b821ab79
fix bug of loss nan (#159)
* fix bug of loss nan

* update

* update ctc loss
2021-05-11 16:13:56 +08:00
kang sheng e7f27ae317
change argument names according to convention (#131)
* change argument names according to convention

* bug fix when rename leakyRelu
2021-05-11 16:07:36 +08:00
jeffreykuang 47896a3f80
Unit tests of dbnet transform (#163)
* Add unit test for dbnet transforms

* Add unit test for dbnet transforms

* Add unit test for dbnet transforms
2021-05-11 15:21:14 +08:00
Hongbin Sun 892d486d01
add model for chinese (#156)
* add model for chinese

* update readme

* update readme

* fix link error; add vis for Chinese
2021-05-09 21:49:08 +08:00
lizz 5054a3f78d
Implement iter protocol for Loader (#150)
* Implement iter protocal for Loader

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

* Test iter protocal

Signed-off-by: lizz <lizz@sensetime.com>
2021-05-06 09:51:29 +00:00
lizz 47965516f0
Improve KIE (#153)
* Improve KIE
* allow data do not contain label
* Make relation float32

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

* Add test

Signed-off-by: lizz <lizz@sensetime.com>
2021-05-06 09:51:10 +00:00
lizz 0e70f99f4d
Improve some writings (#142)
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-30 11:09:04 +08:00
quincylin1 961c450df7
show bounding box in img_demo output (#144) 2021-04-30 11:07:54 +08:00
lizz 45fdc66f11
Add sort_vertex8 helper function (#145)
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-30 11:07:05 +08:00
quincylin1 40f10cc996
fix issue#127: changed save_path to out_file (#141) 2021-04-29 15:46:46 +08:00
Jianyong Chen e29b5b8abe
Fix issue 122 (#130)
* fix #122: textsnake targets adaptation

* fix #122: textsnake targets adaptation

* add unittest

* fix format

* fix textsnake unittest on cpu

* fix unit test coverage

* add unit test
2021-04-29 11:02:57 +08:00
quincylin1 e7847bd8b9
Visualise toy (#137)
* added img_norm_cfg key-value pair for toy data visualization

* add ori_filename
2021-04-28 12:42:08 +00:00
quincylin1 20cda15bcd
add img_norm_cfg to config (#136) 2021-04-28 17:56:00 +08:00
quincylin1 b86609dcf9
added img_norm_cfg key-value pair for toy data visualization (#134) 2021-04-28 17:05:54 +08:00
LvTianlei 7bf88d0e25
Add TPS (#117)
* Add TPS

* Update tps_preprocessor.py

* Add licence, change variable name and format

* renamed some parameters and add tests of ocr preprocessor

* renamed params

* Update tps_preprocessor.py

* add config file and readme for TPS
2021-04-28 07:39:13 +00:00
quincylin1 f97f770941
add mmgen (#123) 2021-04-25 13:52:45 +08:00
Hongbin Sun 1cd3c8d3e2
fix some from pylint (#121) 2021-04-23 13:56:23 +08:00
Hongbin Sun d541de77c7
use mmcv dump instead (#120) 2021-04-23 10:27:12 +08:00
Hongbin Sun 43dcb32d4f
Hbsun/end2end demo (#105)
* add end2end demo

* fix typo

* pad box

* fix bug of crnn

* fix polygon

* update docstring

* fix bug of polygon

* updare demo api

* fix except

* rename

* fix with comments
2021-04-22 20:42:42 +08:00
Sergio Bugallo Enjamio ce98d23ba2
add libGL1-mesa-glx to dockerfile (#109) 2021-04-22 00:24:25 +08:00
Jianyong Chen 0bfda842c9
fix #101: fix assertion (#103)
* fix #101: fix assertion

* fix #101: fix pytest
2021-04-20 17:09:09 +08:00
seekingdeep 6c1c34fb91
increase the number of candidates for DBNet to 3000 (#98)
This solves the issue when dealing with documents with a large number of words per-page.
2021-04-20 10:18:56 +08:00
Hongbin Sun b058fdcb4e
mv data_convert_util to mmocr (#96)
* mv data_convert_util to mmocr

* update

* rm bracket
2021-04-19 21:03:52 +08:00
lizz bba33a3f4f
Support windows (#89)
* Fix compile error for Chinese comments in windows

* Fix pan on windows

* Fix types

* Update setup.py

Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-04-19 11:26:00 +00:00
quincylin1 997a89250e
added qr code to README_zh-CN (#95)
* added qr code to README_zh-CN

* Update setup.cfg

Co-authored-by: lizz <innerlee@users.noreply.github.com>
2021-04-19 18:48:15 +08:00
Hongbin Sun 360082635b
fix #93: update ic17 download url (#94) 2021-04-19 13:29:36 +08:00
Hongbin Sun f5c385b02c
fix inference bug (#87)
* fix inference bug

* add unit test for det
2021-04-19 02:07:25 +00:00
Sergio Bugallo Enjamio a68de2214a
Add support for numpy array inference (#74)
* Add support for numpy arrays in model_inference

* Add test for numpy ndarray inference

* Fix linting problems

* Simplify assertion in model_inference
2021-04-18 20:10:17 +08:00
Hongbin Sun 096c7c11b3
update mmdet install (#84)
* update mmdet install

* update install.md

* fix version of mmcv and mmdet
2021-04-17 22:48:15 +08:00
quincylin1 dd4e19f9bd
added link to readme.md (#80) 2021-04-16 17:45:20 +08:00
quincylin1 3944394ec8
fix pr#70 based on ZZL's comments (#77) 2021-04-16 14:15:45 +08:00
RangiLyu b04cb29037
Fix typo in resnet31_ocr (#71) 2021-04-15 10:25:15 +08:00
Hongbin Sun 1a129a1e98
add svt converter (#65)
* add svt converter

* fix str fmt

* fix str fmt

* update convert script
2021-04-14 18:33:14 +08:00
Hongbin Sun 524498479f
add unit test for textsnake (#61) 2021-04-13 21:59:19 +08:00
Hongbin Sun 344cc9a38d
Add unittest for kie (#56)
* add unittest for sdmgr

* fix #36: update datasets.md

* unittest pass
2021-04-13 13:58:31 +08:00
quincylin1 a019058503
fix #51: revised readme.md for dbnet (#54)
* fix #51: revised readme.md for dbnet

* Update configs/textdet/dbnet/README.md

Co-authored-by: jeffreykuang <kuangzhh@gmail.com>
2021-04-13 11:53:57 +08:00
lizz 5d95d5c0bb
Add cite info (#47)
* Add cite info

* Update README.md

* Update README.md

Co-authored-by: jeffreykuang <kuangzhh@gmail.com>

Co-authored-by: jeffreykuang <kuangzhh@gmail.com>
2021-04-11 13:46:21 +08:00
jeffreykuang 666ac1ff1c
Merge pull request #46 from open-mmlab/innerlee-patch-1
Update README.md
2021-04-10 20:57:04 +08:00
lizz c484627888
Update README.md 2021-04-10 20:44:20 +08:00
jeffreykuang 0dd23bca21
Merge pull request #43 from open-mmlab/zz/gpuci
Enable gpu ci
2021-04-10 20:35:07 +08:00
lizz 579604f563
Update build.yml 2021-04-10 20:03:15 +08:00
jeffreykuang 4bc0f73eb2
Merge pull request #45 from open-mmlab/zz/readme
Add methods to readme
2021-04-10 18:12:17 +08:00
lizz 7dc5695d6d Add methods to readme
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 18:04:36 +08:00
lizz 76b196a069 torch 1.8 and python 3.6, 3.8
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-10 17:28:10 +08:00
lizz 71dbb8a087
Merge pull request #44 from cuhk-hbsun/hbsun/syn90k
fix name mistake
2021-04-10 17:13:31 +08:00
Hongbin Sun 1176b485fa fix name mistake 2021-04-10 17:10:50 +08:00