b7a4eac65f
[Enhancement] Add deprecation message for deploy tools ( #801 )
...
* Add deprecation message for deploy tools
* move wwarning info to start
2022-02-28 16:22:47 +08:00
91f98bc645
[Enhancement] Add open-mmlab precommit hook ( #787 )
2022-02-22 12:52:04 +08:00
c466e6f82d
[Enhancement] Upgrade isort ( #771 )
...
* [Enhancement] Upgrade isort
* Fix files
2022-02-09 23:22:14 +08:00
b054f609b4
[Enhancement] Remove unnecessary custom_import from test.py ( #758 )
2022-01-27 11:02:16 +08:00
e0ff11819b
[Feat] Support specify gpu for testing and training with gpu-id instead of gpu-ids and gpus ( #756 )
...
* support set gpu number for testing and deprecate gpu_ids and gpus with gpu_id
* update gpu-ids gpus and gpu-id dostring
* add args.gpus process
* fix typo
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-01-27 10:11:14 +08:00
d871eed79e
[Enhancement] Add setup multi-processing both in train and test. ( #757 )
2022-01-26 20:34:19 +08:00
218f9f08d4
[Fix] Use yaml.safe_load instead of load ( #753 )
2022-01-26 14:29:30 +08:00
ad3957bb41
[Enhancement] Speed up training ( #739 )
2022-01-19 20:14:41 +08:00
2f429d5e40
Extend totaltext converter to support text fields ( #728 )
...
* Extend totaltext converter to support text fieldols/
* fix bug
* fix comment typo
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2022-01-14 16:00:53 +08:00
c736989615
[Feature] Extend ctw1500 converter to support text fields ( #729 )
...
* Extend ctw1500 converter to support text fieldols/
* remove args for debug
2022-01-14 15:30:48 +08:00
8c5e8c240e
[Enhancement] Disable multi-processing feature of cv2 to speed up data loading ( #721 )
2022-01-11 12:52:19 +08:00
96beda6f2d
[Docs] Add config tutorial ( #683 )
...
* Add config tutorial and reorganize dataset_type.md
* update cn docs toc
* Add misc tool
* Improve docs api menu and config docs
2021-12-28 12:41:24 +08:00
fb1892a1ae
[Fix] fix bug of show_results in single_gpu_test ( #667 )
...
* move single_gpu_test to mmocr/apis
* fix show result in single_gpu_test
* add unittest
* add docstring
* add unittest for kie
* avoid re-define
* remove skipif
* add skip
* fix conflict
* fix with comment
* add note in readme
* update
2021-12-23 03:30:01 +00:00
f8dfbd4177
[Fix] Support both ConcatDataset and UniformConcatDataset ( #675 )
...
* support UniformConcatDataset
* update
* rm useless
* handle 2d-list datasets
2021-12-22 12:32:02 +00:00
abbae7bfd1
fix sartrn onnxruntime test ( #679 )
...
* fix satrn test
* disable aug test for deployment test
2021-12-21 13:52:11 +08:00
a4237ad568
[Enhancement] Support muitiple pipelines for different datasets ( #657 )
...
* support multi pipelines
* support different pipelines
* fix
* fix replace_image_to_tensor
* remove redundant check
* update
* add copyright
* not in-place modify
2021-12-14 12:29:43 +00:00
bdbeb69076
[Fix] Remove depreciated image sanity check ( #661 )
2021-12-10 12:50:41 +08:00
1544e9bb29
[Enhancement] Remove unnecessary custom_import from train.py ( #603 )
2021-11-25 15:59:35 +08:00
e267d06281
[Fix] Always broadcast a random seed to all the processes ( #600 )
2021-11-18 22:26:21 +08:00
98c5bff1e0
[Feature] Save results to json file for kie. ( #589 )
...
* save results json for kie
* update config
* Fix KIE inference bug
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-15 20:28:13 +08:00
4870ca91fc
[Fix] Fix resource warning: unclosed file ( #577 )
...
* fix resource warning: unclosed file
* fix import
* Eliminate det_test_img's dependency on mmdet
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
2021-11-12 12:36:57 +08:00
cae080734f
fix recog_test_imgs ( #574 )
2021-11-11 17:39:54 +08:00
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
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
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
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
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
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
d683b14283
[Fix] Totaltext_converter: skip invalid annotations ( #438 )
...
* [Fix] Skip invalid annoataions
2021-08-20 11:23:05 +08: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
b8f7ead74c
[Enhancement] Add copyright info ( #439 )
...
* add copyright info
2021-08-17 17:39:30 +08: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
969c65165f
[Fix] Remove depreciated pretrained in kie_test_imgs ( #381 )
2021-07-21 15:18:59 +08:00
4f7270e574
Fix #282 : Support init_cfg & update depreciated configs ( #365 )
...
* update coco ref
* init_cfg for dbnet
* initcfg for mask_rcnn
* textsnake init_cfg
* fix dbnet
* panet initcfg
* psenet initcfg
* fcenet initcfg
* drrg initcfg
* add init_cfg to detectors
* update maskrcnn config file to support mmdet
* fix init_cfg of fce_head
* crnn initcfg
* init_weights in training
* nrtr initcfg
* robust_scanner initcfg
* sar init_cfg
* seg init_cfg
* tps_crnn init_cfg
* sdmgr initcfg
* ner init_cfg
* fix textsnake
* sdmgr initcfg
* move "pretrained" to "init_cfg" for config files
* Moduleslist update
* fix seg
* ner init_cfg
* fix base
* fix encode decode recognizer
* revert dbnet config
* fix crnn
* fix base.py
* fix robust_scanner
* fix panet
* fix test
* remove redundant init_weights() in fcehead
* clean up
* relex mmdet version in workflow
* Add dependency version check
* Update mmocr/models/textdet/dense_heads/pse_head.py
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-07-20 23:18:25 +08:00
884755d05d
Fix #112 : Remove the need of drop_orientation_info in data preprocessing steps ( #375 )
...
* ctw1500 ignore orientation
* restore maskrcnn config
* ignore_orientation support for icdar datasets
* update docs
* ignore orientation for total text
* Add LoadOCRImageFromFile
* Fix typo
* simplify design
* remove LoadOCRImageFromFile
* update chinese docs
2021-07-20 23:02:25 +08:00
02e3b98684
fix syntext_converter ( #361 )
2021-07-12 02:07:50 +00:00
243f47dc03
add totaltext for recog and det ( #357 )
...
* add totaltext for recog and det
* add setup
* fix doc
* fix based on comments
2021-07-08 21:52:50 +08:00
68df4fbe80
[Feature] Add synthtext converter and update docs ( #351 )
...
* Add synthtext converter and update docs
* minor docs fix
2021-07-07 15:54:29 +08:00
19aefa1ae1
Added support for ConcatDataset with two workflows ( #348 )
2021-07-06 16:37:37 +08:00
97ac619e68
[Fix] Fix unit tests, remove temporary files ( #328 )
...
* fix unit tests, remove temp files
* use tempfile package and fix typo
* all files clear
2021-06-26 22:51:50 +08:00
8999c56159
Fix kie_test_imgs script for novisual ( #319 )
...
* Fix kie_test_imgs script for novisual
Signed-off-by: lizz <lizz@sensetime.com>
* Fix test
Signed-off-by: lizz <lizz@sensetime.com>
2021-06-24 07:22:34 +00:00
82f64a5b62
support batch inference during testing ( #310 )
...
* support batch inference during testing
* fix unittest
* update docs using url
* set cfg for train, val and test
* update docs
* update docs and test.py
* samples_per_gpu as global setting
* changes revert
2021-06-23 11:34:29 +08:00
e6cb750922
add TextOCR dataset converter ( #293 )
...
* textocr converter for text recog
* textocr converter for text detection
* update documentation
* remove unnecessary garbage collection lines
* multi-processing textocr converter
* json->mmcv, fix documentation
2021-06-21 03:06:10 +00:00
f1b003ddb1
[Feature] Add deployment evaluation ( #291 )
...
* add deployment evaluation
* fix lint
* remove cpu unit tests for trt and onnx
* use pytest.mark to skip cpu unit test
* move to mmocr/core
* emm... renamed to wrappers
* renamed to deploy_utils
* renamed unit test to test_deploy_utils
* fix lint
* using pytest.mark.importorskip
2021-06-18 12:27:10 +08:00
xinke-wang
Tong Gao
liukuikun
Hongbin Sun
AllentDan
Darwin Bautista
liukuikun
lizz
quincylin1
Antonio Lanza