Ma Zerun
b8b31e9343
[Improve] Multiple fixes. ( #957 )
...
* Rename some files in unit tests.
* Rename `pipelines` folder to `transforms`.
* Fix version requirement range.
* Remove `init_random_seed` and `set_random_seed`.
* Fix unit tests
* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00
mzr1996
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
mzr1996
dd660ed99e
[Improve] Clean useless code and reduce unit tests memory usage.
2022-07-18 11:11:13 +08:00
Ezra-Yu
d6fa480915
[Refactor] Add VOC dataset and mutil_label dataset
2022-07-18 11:11:13 +08:00
mzr1996
efaa93aae1
[Refactor] Refactor entry scripts `train.py` and `test.py`.
2022-07-18 10:53:56 +08:00
mzr1996
27e685fe10
[Feature] Add `BaseDataset`, `CustomDataset`, `ImageNet` and `ImageNet21k`
2022-07-18 10:53:27 +08:00
Ma Zerun
d0d6f73fab
[Feature] Add `CustomDataset`. ( #738 )
...
* Add custom dataset and refactor ImageNet dataset
* Add default CLASSES for CIFAR dataset
* Add unit tests
* Imporve according to comments
2022-03-30 15:28:26 +08:00
takuoko
aa522f4309
[Feature] Support CUB dataset. ( #703 )
...
* support cub dataset
* support cub dataset
* fix train lint error
* add docs
* fix class label
Co-authored-by: Ezra-Yu <1105212286@qq.com>
* del debug code
* skip docformatter problem
* add unit tests
* add CUB baseline configs and chpts
* fix some typos
* fix name style
* update flops
Co-authored-by: Ezra-Yu <1105212286@qq.com>
2022-03-16 16:22:28 +08:00
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
d5ae255061
[Feature] Repeated Aug and Sampler Registry ( #588 )
2021-12-21 15:26:28 +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
imyhxy
671414becb
[Fix] Fix missing import `Compose` and `Normalize`.
...
* Fixed missing import 'Compose'
* Fixed mistype `Compose` in `mmcls/datasets/__init__.py`
* Fixed missing import `Normalize`
* [Docs] Fix typos in doctest
* [Fix] Sort import module
2021-10-28 15:21:05 +08:00
Ma Zerun
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
LXXXXR
c07e60dd7d
[Feature] Support PASCAL VOC 2007 dataset for multilabel task ( #134 )
...
* support voc
* minor change
2021-01-11 11:08:59 +08:00
yanglei
ba789da5c8
Add commonly used datasets
2020-07-01 16:09:06 +08:00
chenkai
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00