94cd52dc4a
simplify installation procedure ( #188 )
...
* postproc
* mv postproc of textdet in mmcv
* linting
* change deps mmcv 1.3.4
* update deps
* linting
* Update configs/textdet/panet/panet_r18_fpem_ffm_600e_icdar2015.py
* Update docs/changelog.md
* Update mmocr/models/textdet/postprocess/__init__.py
* Update mmocr/version.py
* resize&recompress jpg
* Update changelog.md
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-15 10:34:37 +08:00
cbdd98a1e1
add fcenet ( #133 )
...
* add fcenet
* fix linting and code style
* fcenet finetune
* Update transforms.py
* Update fcenet_r50dcnv2_fpn_1500e_ctw1500.py
* Update fcenet_targets.py
* Update fce_loss.py
* fix
* add readme
* fix config
* Update fcenet_r50dcnv2_fpn_1500e_ctw1500.py
* fix
* fix readme
* fix readme
* Update test_loss.py
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-14 21:37:04 +08:00
1240169f18
update metafile ( #183 )
2021-05-14 00:17:33 +08:00
05aedf11d6
update demo docs for batch inference ( #181 )
2021-05-13 17:58:38 +08:00
7e0d48df11
fix metafiles ( #182 )
2021-05-13 17:28:59 +08:00
97e2f27017
fix #173 : support aug test ( #178 )
...
* fix #173 : support aug test
* fix pytest
* support batch inference for both text det and recog
* update unittest
* use one img for batch infer test
2021-05-13 07:18:00 +00:00
8e65bea9c3
Add mminstall ( #180 )
...
* add mminstall.txt
* fix mminstall.txt
* fix mminstall
2021-05-13 15:06:12 +08:00
4b17d43bb0
add manifest.in ( #175 )
2021-05-13 03:16:10 +00:00
cc27d8254b
Add metafiles ( #179 )
...
* add metafiles
* add metafiles
* add metafiles
* fix metafile errors
* fix metafiles
* Update metafile.yml
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-13 02:46:26 +00:00
ee700e94cd
add mminstall.txt ( #177 )
2021-05-13 09:01:31 +08:00
ca1ddfd6fd
add include_package_data to setup.py ( #176 )
2021-05-12 22:55:50 +08:00
e2643df439
move demo image to demo/resources ( #171 )
...
* move demo image to demo/resources
* update index.rst
* revert
* update github link for demo image
2021-05-12 20:35:39 +08:00
9be85faa82
Add things from "import *" to __all__ ( #167 )
...
* Add things from "import *" to __all__
Signed-off-by: lizz <lizz@sensetime.com>
* Remove tile_image
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-12 20:22:30 +08:00
04273b325a
Update image link ( #170 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-12 06:16:22 +00:00
df5493a79e
refactor tps config ( #135 )
...
* refactor tps config
* recover tps
* add ckpt of tps
2021-05-12 06:14:24 +00:00
18c54afbdc
Add support for batch inference ( #86 )
...
* Add support for numpy arrays in model_inference
* Add test for numpy ndarray inference
* Fix linting problems
* Add support for batch inference
* Add batch inference demo script
* Fix comment
* Test batch inference with paths and arrays
* lint code
* Update model_inference docstring
* Refactor model inference tests
* Change inference function to make text detectors and recognizers use the same input data types
* Change single state text detector model to support batch inference
* Lint code
* simplify inference tests
* Remove psenet from batch inference test cases to prevent the pytest being killed
* Update batch_image_demo.py
* fix bug when test with dataset
fix bug when test with dataset, for example, `./tools/dist_test.sh configs/textrecog/sar/sar_r31_parallel_decoder_academic.py <checkpoint> 1 --eval acc`
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-05-12 02:04:01 +00:00
3b9306ca79
Hbsun/demo doc ( #168 )
...
* move demo.md to demo/docs
* update link in md
2021-05-12 09:57:53 +08:00
45d22b12c6
update demo docs ( #166 )
...
* update demo docs
* update docs
2021-05-11 21:13:19 +08:00
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
6a1d1020f6
Add a message when annotation is empty ( #165 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-11 16:38:30 +08:00
85b821ab79
fix bug of loss nan ( #159 )
...
* fix bug of loss nan
* update
* update ctc loss
2021-05-11 16:13:56 +08:00
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
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
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
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
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
0e70f99f4d
Improve some writings ( #142 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-30 11:09:04 +08:00
961c450df7
show bounding box in img_demo output ( #144 )
2021-04-30 11:07:54 +08:00
45fdc66f11
Add sort_vertex8 helper function ( #145 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-04-30 11:07:05 +08:00
40f10cc996
fix issue#127: changed save_path to out_file ( #141 )
2021-04-29 15:46:46 +08:00
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
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
20cda15bcd
add img_norm_cfg to config ( #136 )
2021-04-28 17:56:00 +08:00
b86609dcf9
added img_norm_cfg key-value pair for toy data visualization ( #134 )
2021-04-28 17:05:54 +08:00
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
f97f770941
add mmgen ( #123 )
2021-04-25 13:52:45 +08:00
1cd3c8d3e2
fix some from pylint ( #121 )
2021-04-23 13:56:23 +08:00
d541de77c7
use mmcv dump instead ( #120 )
2021-04-23 10:27:12 +08:00
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
ce98d23ba2
add libGL1-mesa-glx to dockerfile ( #109 )
2021-04-22 00:24:25 +08:00
0bfda842c9
fix #101 : fix assertion ( #103 )
...
* fix #101 : fix assertion
* fix #101 : fix pytest
2021-04-20 17:09:09 +08:00
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
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
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
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
360082635b
fix #93 : update ic17 download url ( #94 )
2021-04-19 13:29:36 +08:00
f5c385b02c
fix inference bug ( #87 )
...
* fix inference bug
* add unit test for det
2021-04-19 02:07:25 +00:00
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
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
dd4e19f9bd
added link to readme.md ( #80 )
2021-04-16 17:45:20 +08:00
jeffreykuang
Zyq-scut
Hongbin Sun
quincylin1
lizz
Sergio Bugallo Enjamio
kang sheng
Jianyong Chen
LvTianlei
seekingdeep
Sergio Bugallo Enjamio