361 Commits

Author SHA1 Message Date
Ezra-Yu
e9342d9e4c Add multi_label heads 2022-07-18 11:11:13 +08:00
yingfhu
4fcd7ee072 Add EfficientNetRandomCrop and EfficientNetCenterCrop 2022-07-18 11:11:13 +08:00
mzr1996
12c982f939 [Refactor] Refactor RepeatAugSampler. 2022-07-18 11:11:13 +08:00
yingfhu
e8d69cf2ff [Fix] adjust default dataset path to empty string 2022-07-18 11:11:13 +08:00
Ezra-Yu
7bca2516f0 Add CUB dataset 2022-07-18 11:11:13 +08:00
Ezra-Yu
daa6167336 Add Lighting, ColorJitter and Albumentations 2022-07-18 11:11:13 +08:00
Ezra-Yu
a9057e88c4 Add multi label metrics 2022-07-18 11:11:13 +08:00
yingfhu
62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
Ezra-Yu
a82de04b67 Enhance browse datset 2022-07-18 11:11:13 +08:00
mzr1996
daa716b112 Add some type hints 2022-07-18 11:11:13 +08:00
mzr1996
69e5ab065e Change "Default to" to "Defaults to" 2022-07-18 11:11:13 +08:00
mzr1996
7f13507fa8 Revert LongTensor check since cuda tensor is not long tensor 2022-07-18 11:11:13 +08:00
mzr1996
375fe68f12 Imporve according to comments 2022-07-18 11:11:13 +08:00
mzr1996
f0cab33e09 Force label in ClsDatasample long type. 2022-07-18 11:11:13 +08:00
mzr1996
f3299b4ca2 [Refactor] Refactor batch augmentations 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
mzr1996
cecff79a79 Rename 'feat' mode to 'tensor' mode 2022-07-18 11:11:13 +08:00
yingfhu
125b74d4ca [Refactor] Refactor MNIST and FashionMNIST dataset. 2022-07-18 11:11:13 +08:00
mzr1996
ee6e585e41 Mazerun/base classifier 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
522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +08:00
yingfhu
c78b5597d8 [Refactor] refactor randomCrop, randomResizeCrop and CenterCrop。 2022-07-18 10:57:17 +08:00
mzr1996
05e4c13343 Improve according to comments 2022-07-18 10:53:56 +08:00
mzr1996
c0feadf546 Refactor RandomErasing 2022-07-18 10:53:56 +08:00
mzr1996
4f28b9dd63 [Refactor] refactor AutoAugment and RandAugment 2022-07-18 10:53:56 +08:00
mzr1996
548db6f4ac [Refactor] Update optimizer related registries and configs. 2022-07-18 10:53:56 +08:00
mzr1996
3b6d44ea05 [Refactor] Refactor CIFAR10 and CIFAR100 dataset. 2022-07-18 10:53:56 +08:00
mzr1996
995b1d0d58 [Refactor] Add ResizeEdge and refactor all dataset configs. 2022-07-18 10:53:56 +08:00
Ezra-Yu
4d4b22377d Add browse dataset 2022-07-18 10:53:56 +08:00
mzr1996
efaa93aae1 [Refactor] Refactor entry scripts train.py and test.py. 2022-07-18 10:53:56 +08:00
mzr1996
6ad75f0076 [Feature] Add metrics for single-label classification. 2022-07-18 10:53:28 +08:00
Ezra-Yu
93a27c8324 [Feature] Add PackClsInputs and use LoadImageFromFile, Resize & RandomFlip in MMCV. 2022-07-18 10:53:28 +08:00
mzr1996
0537c4d70c [Feature] Add ClsVisualizer. 2022-07-18 10:53:27 +08:00
mzr1996
27e685fe10 [Feature] Add BaseDataset, CustomDataset, ImageNet and ImageNet21k 2022-07-18 10:53:27 +08:00
mzr1996
98377df512 Update ClsDataSample design 2022-07-18 10:53:27 +08:00
mzr1996
ce2b40133b Add property, setter and deleter. 2022-07-18 10:53:27 +08:00
mzr1996
4bf54066c9 Add ClsDataSample 2022-07-18 10:53:27 +08:00
yingfhu
de002e455f [Refactor] Use MMLogger from MMEngine in get_logger and print_log. 2022-07-18 10:53:27 +08:00
Ezra-Yu
088d5b5add [Refactor] Inherit all registries from MMEngine and use inherited registries. 2022-07-18 10:53:27 +08:00
Ma Zerun
313d35704f
Bump version to v0.23.1 (#863) 2022-06-02 21:22:54 +08:00
Ayush Thakur
ccdbc82e39
[Feature] Dedicated MMClsWandbHook for MMClassification (Weights and Biases Integration) (#764)
* wandb integration

* visualize using wandb tables

* wandb tables enhanced

* Refactor MMClsWandbHook (#1)

* [Enhance] Add extra dataloader settings in configs. (#752)

* Use `train_dataloader`, `val_dataloader` and `test_dataloader` settings
in the `data` field to specify different arguments.

* Fix bug

* Fix bug

* [Enhance] Improve CPE performance by reduce memory copy. (#762)

* [Feature] Support resize relative position embedding in `SwinTransformer`. (#749)

* [Feature]: Add resize rel pos embed

* [Refactor]: Create a separated resize_rel_pos_bias_table func

* [Refactor]: Refactor rel pos embed bias

* [Refactor]: Move interpolate into func

* Remove index buffer only when window_size changes

Co-authored-by: mzr1996 <mzr1996@163.com>

* [Feature] Add PoolFormer backbone and checkpoints. (#746)

* add PoolFormer

* fix some typos in PoolFormer

* fix lint error

* modify out_indices and gap

* fix typo

* fix lint

* fix typo

* fix typo in poolforemr README

* fix lint

* Update some paths

* Refactor freeze_stages method

* Add unit tests

* Fix lint

Co-authored-by: mzr1996 <mzr1996@163.com>

* Bump version to v0.22.1 (#785)

* [Docs] Refine API reference. (#774)

* [Docs] Refine API reference

* Add PoolFormer

* [Docs] Fix docs.

* [Enhance] Reduce the memory usage of unit tests for Swin-Transformer. (#759)

* [Feature] Support VAN. (#739)

* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>

* [Feature] Support DenseNet. (#750)

* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>

* [Fix] Use symbolic link in the API reference of Chinese docs.

* [Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723)

* implement training and evaluation on IPU

* fp16 SOTA

* Tput reaches 5600

* 123

* add poptorch dataloder

* change ipu_replicas to ipu-replicas

* add noqa to config long line(website)

* remove ipu dataloder test code

* del one blank line in test_builder

* refine the dataloder initialization

* fix a typo

* refine args for dataloder

* remove an annoted line

* process one more conflict

* adjust code structure in mmcv.ipu

* adjust ipu code structure in mmcv

* IPUDataloader to IPUDataLoader

* align with mmcv

* adjust according to mmcv

* mmcv code structre fixed

Co-authored-by: hudi <dihu@graphcore.ai>

* [Fix] Fix lint and mmcv version requirement for IPU.

* Bump version to v0.23.0 (#809)

* Refacoter Wandb hook and refine docstring

Co-authored-by: XiaobingZhang <xiaobing.zhang@intel.com>
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Weihao Yu <1090924009@qq.com>
Co-authored-by: takuoko <to78314910@gmail.com>
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com>
Co-authored-by: Hu Di <476658825@qq.com>
Co-authored-by: hudi <dihu@graphcore.ai>

* shuffle val data

* minor updates

* minor fix

Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: XiaobingZhang <xiaobing.zhang@intel.com>
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Weihao Yu <1090924009@qq.com>
Co-authored-by: takuoko <to78314910@gmail.com>
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com>
Co-authored-by: Hu Di <476658825@qq.com>
Co-authored-by: hudi <dihu@graphcore.ai>
2022-06-02 17:58:49 +08:00
Ma Zerun
3e198e63e8
[Docs] Update install tutorials. (#854)
* [Docs] Update install tutorials.

* [Docs] Improve dataset docs

* Add option to show the results in demo.

* fix typo
2022-06-01 18:31:57 +08:00
Ma Zerun
7c5ddb1e5b
Bump version to v0.23.0 (#809) 2022-05-01 21:58:33 +08:00
mzr1996
1d6fbe0efe [Fix] Fix lint and mmcv version requirement for IPU. 2022-04-29 22:33:29 +08:00
Hu Di
b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723)
* implement training and evaluation on IPU

* fp16 SOTA

* Tput reaches 5600

* 123

* add poptorch dataloder

* change ipu_replicas to ipu-replicas

* add noqa to config long line(website)

* remove ipu dataloder test code

* del one blank line in test_builder

* refine the dataloder initialization

* fix a typo

* refine args for dataloder

* remove an annoted line

* process one more conflict

* adjust code structure in mmcv.ipu

* adjust ipu code structure in mmcv

* IPUDataloader to IPUDataLoader

* align with mmcv

* adjust according to mmcv

* mmcv code structre fixed

Co-authored-by: hudi <dihu@graphcore.ai>
2022-04-29 22:22:19 +08:00
Hubert
59292b315a
[Feature] Support DenseNet. (#750)
* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-28 13:06:03 +08:00
takuoko
df6edd7f5a
[Feature] Support VAN. (#739)
* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>
2022-04-28 09:35:17 +08:00
Ma Zerun
702c196514
[Docs] Refine API reference. (#774)
* [Docs] Refine API reference

* Add PoolFormer
2022-04-15 20:19:20 +08:00
mzr1996
29b882d4eb Merge branch 'dev' 2022-04-15 20:10:45 +08:00
Ma Zerun
eda75d12ec
Bump version to v0.22.1 (#785) 2022-04-15 20:08:39 +08:00