mmpretrain/tools
Ma Zerun b4ee9d2848
[Feature] Support calculate confusion matrix and plot it. ()
* [Feature] Support calculate confusion matrix and plot it.

* Fix keepdim

* Update confusion_matrix tools and the plot graph.

* Revert accidental modification.

* Update docstring

* Move confusion matrix tool to
2023-02-14 12:58:11 +08:00
..
analysis_tools [Feature] Support calculate confusion matrix and plot it. () 2023-02-14 12:58:11 +08:00
misc [Docs] update visualization doc. () 2022-11-18 12:44:55 +08:00
model_converters [Feature] Support LeViT backbone. () 2023-01-17 17:43:42 +08:00
torchserve [Fix] Fix the torchserve. () 2022-11-21 11:18:42 +08:00
visualizations [Improve] Update registries of mmcls. () 2023-01-11 15:20:51 +08:00
dist_test.sh [Feature] Add multi machine dist_train () 2022-03-23 12:03:03 +08:00
dist_train.sh [Feature] Add multi machine dist_train () 2022-03-23 12:03:03 +08:00
kfold-cross-valid.py [Improve] Update registries of mmcls. () 2023-01-11 15:20:51 +08:00
slurm_test.sh init commit 2020-05-21 21:21:43 +08:00
slurm_train.sh init commit 2020-05-21 21:21:43 +08:00
test.py [Refactor] Unify the `--out` and `--dump` in `tools/test.py`. () 2023-02-09 14:05:03 +08:00
train.py [Improve] Update registries of mmcls. () 2023-01-11 15:20:51 +08:00