mmclassification

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	b39885d953	[Feature] Support K-fold cross-validation (#563 ) * Support to use `indices` to specify which samples to evaluate. * Add KFoldDataset wrapper * Rename 'K' to 'num_splits' accroding to sklearn * Add `kfold-cross-valid.py` * Add unit tests * Add help doc and docstring	2022-01-19 18:32:55 +08:00
Zhicheng Chen	da39ca6898	[Fix] Fix test sampler bug. (#611 ) * sampler bugfixes * sampler bugfixes * reorganize code * minor fixes * reorganize code * minor fixes * Use `mmcv.runner.get_dist_info` instead of `dist` package to get rank and world size. * Add `build_dataloader` unit tests and fix sampler's unit tests. * Fix unit tests * Fix unit tests Co-authored-by: mzr1996 <mzr1996@163.com>	2021-12-23 18:07:41 +08:00
Zhicheng Chen	d5ae255061	[Feature] Repeated Aug and Sampler Registry (#588 )	2021-12-21 15:26:28 +08:00
Ezra-Yu	4b2128fdbd	[Fix] Fix bug when using `ClassBalancedDataset` (#555 ) * fix dataset wrapper bug and add unit tests * fix dataset wrapper bug and add unit tests * update unit tests * fix lint * align interface to mmdet * update unit tests	2021-11-26 09:40:00 +08:00
Ezra-Yu	2ce5825ef1	[Feature] Support ImageNet21k dataset. (#461 ) * add imagnet21k * Update unit test * update imaenet21k * use slots * use slots * rename Data_item to ImageInfo * add unit tests * Update unit tests * rm some print * update unit tests * fix lint * remove default value of pipeline	2021-10-28 15:22:08 +08:00
Ma Zerun	2e6c7cf87d	[Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470 ) * Add code spell check hook * Add codespell config * Fix a lot of typos. * Add formating.py to keep compatibility.	2021-10-13 14:33:07 +08:00
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413 )	2021-08-17 19:52:42 +08:00
Ma Zerun	15cd34bbef	[Fix] Use zero as default value of `thrs` in metrics. (#341 ) * Use zero as default value of `thrs` in metrics. And it accepcts a number instead of float now. * Fix unit test comment * Don't pass thrs if no thrs.	2021-07-18 16:57:21 +08:00
Ma Zerun	1a7cebe4b9	[Refactor] Refactor unittest (#321 ) * Refactor unit tests folder structure. * Remove label smooth and Vit test in `test_classifiers.py` * Rename test_utils in dataset to test_dataset_utils * Split test_models/test_utils/test_utils.py to multiple sub files. * Add unit tests of classifiers and heads * Use patch context manager. * Add unit test of `is_tracing`, and add warning in `is_tracing` if torch verison is smaller than 1.6.0	2021-07-08 22:49:05 +08:00

9 Commits (a7f8e96b31c10ab3e9c133293ca406e6e548475b)