mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
977 Commits
5 Branches
20 Tags
54 MiB
Commit Graph

21 Commits (29107ef81da05b3ad0cc8847ae3b8c13501a1716)

Author SHA1 Message Date
Tong Gao 7fcfa09431
[Fix] Support customize runner and visualization in train/test.py, an… (#1279) 
* [Fix] Support customize runner and visualization in train/test.py, and update configs missing from dataflow refactor

* Fix vis

* Apply suggestions from code review

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>

* [Config] Refactor & fix DB, DBPP, DRRG configs (#1181)

* refactor base datasets, fix drrg config

* rename

* update dbnet and drrg

* fix

* fix

* Raise Error

Co-authored-by: Xinyu Wang <45810070+xinke-wang@users.noreply.github.com>
 2022-08-22 10:48:50 +08:00
liukuikun 1b33ff5d76 [Feat] support fp16 auto resume and auto scale lr 2022-07-21 10:58:03 +08:00
liukuikun fe43259a05 [Refactor] train and test 2022-07-21 10:50:59 +08:00
liukuikun a90b9600ce [Refactor] refactor DATASETS and TRANSFORMS 2022-07-21 10:50:55 +08:00
#W[_t 1bd26f24ba
[Fix]: access params by cfg.get (#1145) 
cfg.get is better way to access the params when it may not exist.
 2022-07-11 19:04:05 +08:00
Tong Gao b37b1d05dd
[Fix] Use hyphen instead of underscores in script args (#890) 2022-03-29 10:28:38 +08:00
Tong Gao c25404b358
[Enhancement] Option for different seeds on different ranks (#820) 2022-03-10 10:43:36 +08:00
liukuikun 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
liukuikun d871eed79e
[Enhancement] Add setup multi-processing both in train and test. (#757) 2022-01-26 20:34:19 +08:00
Tong Gao ad3957bb41
[Enhancement] Speed up training (#739) 2022-01-19 20:14:41 +08:00
Tong Gao 8c5e8c240e
[Enhancement] Disable multi-processing feature of cv2 to speed up data loading (#721) 2022-01-11 12:52:19 +08:00
Hongbin Sun 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
Tong Gao 1544e9bb29
[Enhancement] Remove unnecessary custom_import from train.py (#603) 2021-11-25 15:59:35 +08:00
Tong Gao e267d06281
[Fix] Always broadcast a random seed to all the processes (#600) 2021-11-18 22:26:21 +08:00
Tong Gao b8f7ead74c
[Enhancement] Add copyright info (#439) 
* add copyright info
 2021-08-17 17:39:30 +08:00
Tong Gao 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
Antonio Lanza 19aefa1ae1
Added support for ConcatDataset with two workflows (#348) 2021-07-06 16:37:37 +08:00
lizz 491866ea27
Make tools executable (#209) 
Signed-off-by: lizz <lizz@sensetime.com>
 2021-05-18 15:19:08 +08:00
hq_wei 24c590bb04
Ner task (#148) 
* update ner standard code format

* add pytest

* fix pre-commit

* Annotate the dataset section

* fix pre-commit for dataset

* rm big files and add comments in dataset

* rename configs for ner task

* minor changes if metric

* Note modification

* fix pre-commit

* detail modification

* rm transform

* rm magic number

* fix warnings in pylint

* fix pre-commit

* correct help info

* rename model files

* rename err fixed

* 428_tag

* Adjust to more general pipline

* update unit test rate

* update

* Unit test coverage over 90% and add Readme

* modify details

* fix precommit

* update

* fix pre-commit

* update

* update

* update

* update result

* update readme

* update baseline config

* update config and small minor changes

* minor changes in readme and etc.

* back to original

* update toy config

* upload model and log

* fix pytest

* Modify the notes.

* fix readme

* Delete Chinese punctuation

* add demo and fix some logic and naming problems

* add To_tensor transformer for ner and load pretrained model in config

* delete extra lines

* split ner loss to MaskedCrossEntropyLoss and MaskedFocalLoss

* update config

* fix err

* updata

* modify noqa

* update new model report

* fix err in ner demo

* Update ner_dataset.py

* Update test_ner_dataset.py

* Update ner_dataset.py

* Update ner_transforms.py

* rm toy config and data

* add comment

* add empty

* fix conflict

* fix precommit

* fix pytest

* fix pytest err

* Update ner_dataset.py

* change dataset name to cluener2020

* move the postprocess in metric to convertor

* rm __init__ etc.

* precommit

* add discription in loss

* add auto download

* add http

* update

* remove some 'issert'

* replace unsqueeze

* update config

* update doc and bert.py

* update

* update demo code

Co-authored-by: weihuaqiang <weihuaqiang@sensetime.com>
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
 2021-05-18 11:33:51 +08:00
lizz 393ed7fc5a
Implement get_root_logger and train_detector (#4) 2021-04-08 01:24:04 +08:00
jeffreykuang 31f1c007c0 [feature]: add textdet 2021-04-03 01:03:52 +08:00