e9d5b46cb4
When testing my model with tools/test.py, my config will be messed up by the "--save-preds" option. ( #2052 )
...
* fix tools/test.py
* fix tools/test.py
* Update tools/test.py
---------
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
2024-06-02 11:16:52 +08:00
73df26d749
[Enhancement] Accepts local-rank in train.py and test.py ( #1806 )
...
* [Enhancement] Accepts local-rank
* add
* update
2023-03-27 10:34:54 +08:00
1127240108
[Feature] Support auto import modules from registry. ( #1731 )
...
* [Feature] Support auto import modules from registry.
* limit mmdet version
* location parrent dir if it not exist
2023-02-17 10:28:34 +08:00
f820470415
[Feature] Rec TTA ( #1401 )
...
* Support TTA for recognition
* updata readme
* updata abinet readme
* updata train_test doc for tta
2023-02-16 10:27:07 +08:00
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
efd81b7a5a
fix bug
2022-07-21 10:57:17 +08:00
8313e698d2
[Feature] Add offline eval script
2022-07-21 10:51:04 +08:00
fe43259a05
[Refactor] train and test
2022-07-21 10:50:59 +08:00
a90b9600ce
[Refactor] refactor DATASETS and TRANSFORMS
2022-07-21 10:50:55 +08:00
b4a9a87eee
[Enhancement] More customizable fields in dataloaders ( #933 )
...
* [Enhancement] More customizable fields in val and test dataloaders
* update default_loader_cfg
2022-04-18 09:24:07 +08:00
4c57bd35ac
[Docs] Fix the misleading description in test.py ( #908 )
...
* [Docs] Fix misleading description in test.py
* update docs
2022-04-02 11:31:48 +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
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
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
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
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
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
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
d57f279083
[Fix] Correct isort setup ( #288 )
...
* isort
* remove tools from setup.cfg
2021-06-16 19:44:01 +08:00
491866ea27
Make tools executable ( #209 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-05-18 15:19:08 +08:00
44ca9c2a61
Remove usage of \ ( #49 )
...
* Remove usage of \
Signed-off-by: lizz <lizz@sensetime.com>
* rebase
Signed-off-by: lizz <lizz@sensetime.com>
* typos
Signed-off-by: lizz <lizz@sensetime.com>
* Remove test dependency on tools/
Signed-off-by: lizz <lizz@sensetime.com>
* Remove usage of \
Signed-off-by: lizz <lizz@sensetime.com>
* rebase
Signed-off-by: lizz <lizz@sensetime.com>
* typos
Signed-off-by: lizz <lizz@sensetime.com>
* Remove test dependency on tools/
Signed-off-by: lizz <lizz@sensetime.com>
* typo
Signed-off-by: lizz <lizz@sensetime.com>
* KIE in keywords
Signed-off-by: lizz <lizz@sensetime.com>
* some renames
Signed-off-by: lizz <lizz@sensetime.com>
* kill isort skip
Signed-off-by: lizz <lizz@sensetime.com>
* aggregation discrimination
Signed-off-by: lizz <lizz@sensetime.com>
* aggregation discrimination
Signed-off-by: lizz <lizz@sensetime.com>
* tiny
Signed-off-by: lizz <lizz@sensetime.com>
* fix bug: model infer on cpu
Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
2021-04-06 12:16:46 +00:00
31f1c007c0
[feature]: add textdet
2021-04-03 01:03:52 +08:00
Alexnet
Tong Gao
liukuikun
xinyu
liukuikun
Hongbin Sun
lizz
AllentDan
jeffreykuang