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
504e71c3e0
[Enhance] Reduce the memory usage of unit tests for Swin-Transformer. ( #759 )
2022-04-22 17:41:15 +08:00
6beac5060f
[Docs] Fix docs.
2022-04-15 21:25:29 +08:00
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +08:00
29b882d4eb
Merge branch 'dev'
2022-04-15 20:10:45 +08:00
eda75d12ec
Bump version to v0.22.1 ( #785 )
2022-04-15 20:08:39 +08:00
2eb27d0d4e
[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>
2022-04-15 11:35:15 +08:00
fbcf907dc6
[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>
2022-04-13 23:06:56 +08:00
875195efe9
[Enhance] Improve CPE performance by reduce memory copy. ( #762 )
2022-04-02 11:12:20 +08:00
4a3f8d02c5
Merge branch 'master' into dev
2022-04-01 16:51:41 +08:00
5a67bc8a1a
[Fix] Fix unexpected modification on `image_demo.py`
2022-04-01 16:50:23 +08:00
02c8f82147
[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
2022-04-01 15:57:12 +08:00
f0ee5dcb2a
Update checkpoint path of ResNet-50 on CUB dataset
2022-03-31 10:23:33 +08:00
349ec86801
Merge remote-tracking branch 'origin/dev'
2022-03-31 01:37:39 +08:00
2e7f99946b
Bump version to v0.22.0. ( #756 )
2022-03-31 01:36:53 +08:00
922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. ( #709 )
...
* add repmlp
* refactor backbone code
* add checkpoint, readme, unit tests
* unchange demo
* improve docstring.
* update pic
* refactor patch embed
* refactor unit tests
* fix lint
* update tools
* update tools
* Update checkpoint path
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 20:40:25 +08:00
34dd3288ce
[Docs] Add brief installation steps in README for copy&paste. ( #755 )
...
* [Docs] Add breif installation code for copy&paste.
* minor fix
2022-03-30 19:16:45 +08:00
a87ff7e153
[Enhance] Use `--a-b` instead of `--a_b` in arguments. ( #754 )
2022-03-30 19:16:28 +08:00
292127098f
[Feature] Add CSPNet and backbone and checkpoints ( #735 )
...
* add cspnet backbone
* try
* add backbones
* add cspnet
* add acc
* add unit tests
* chnage mode
* Update checkpoint path
* Fix typo
* Imporve CSPNet
* Update checkpoints
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 15:29:35 +08:00
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
995f2912ab
[Feature] Add diff seeds to diff ranks. ( #744 )
...
* [Feature] Add diff seeds to diff ranks.
* lint
2022-03-25 15:31:13 +08:00
1717efadbe
[Fix] Fix the discontiguous output feature map of ConvNeXt. ( #743 )
...
* [Fix] Fix the discontiguous output feature map of ConvNeXt.
* Add comment
2022-03-25 12:34:44 +08:00
a19c28fe95
[Enhance] Add `get_cat_ids` and `get_gt_labels` to KFoldDataset. ( #721 )
2022-03-23 15:57:36 +08:00
84654b17af
[Docs] fix logo url link from mmocr to mmclassification ( #732 )
2022-03-23 15:26:41 +08:00
04cb42a768
[Feature] Support ConvMixer. ( #716 )
...
* basic support for ConvMixer
* simplify
* add data pipeine config for timm
* Add model readme and metafile
* add unittest for convmixer
* add copyright
* modify
* add tests
* update model
* add conv2dAdaptivePadding replacement
* update model index
* fix comments
* Update checkpoint path
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-23 15:23:57 +08:00
3482521587
[Feature] Add multi machine dist_train ( #734 )
...
* add dist_train with multi machines
* add dist_train with multi machines
2022-03-23 12:03:03 +08:00
7856141132
[Enhance] Set torch seed in `worker_init_fn` ( #733 )
2022-03-16 23:12:57 +08:00
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
c1534f9126
[Feature] Support resizemix. ( #676 )
...
* add resizemix
* skip torch.__version__ < 1.7.0
* Update mmcls/models/utils/augment/resizemix.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* Update mmcls/models/utils/augment/resizemix.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* resize -> F.interpolate
* fix docs
* fix test
* add Copyright
* add argument interpolation
Co-authored-by: Ma Zerun <mzr1996@163.com>
2022-03-07 12:11:20 +08:00
2037260ea6
Merge remote-tracking branch 'origin/dev'
2022-03-04 16:13:10 +08:00
bc781ac3cd
Bump version to v0.21.0 ( #722 )
...
* Bump version to v0.21.0
* Update by comment
2022-03-04 16:12:25 +08:00
701b4269c5
[Fix] Add `dist.barrier` to fix a bug in directory checking. ( #666 )
...
* dist barrier instead of sleep
* Update test.py
add time.sleep before dist.barrier
2022-03-04 14:57:08 +08:00
d08c2a148a
[Enhance] Enhance vis-pipeline tool. ( #604 )
...
* enhance vis-pipeline add intermediate imgs
* enhance vis-pipeline add intermediate imgs
* improve code of vi-pipeline
* modify docs for vis-pipeline
* Use `mmcv.utils.digit_version` instead of `distutils`
* add size info in the bottom
* preform adaptive-resize in before concat
* add warning info
* fix docs
* fix lint
* fix comment
* fix docs
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-04 14:40:02 +08:00
779a06257c
[Feature] Support Wide-ResNet. ( #715 )
...
* Add configs of Wide ResNet
* updated config
* add docs and metafile
* update configs
* remove extra import
* Update metafile and readme
Co-authored-by: Ezra-Yu <1105212286@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-04 14:06:34 +08:00
c708770b42
[Enhance] Support dynamic input shape for ViT-based algorithms. ( #706 )
...
* Move `resize_pos_embed` to `mmcls.models.utils`
* Refactor Vision Transformer
* Refactor DeiT
* Refactor MLP-Mixer
* Refactor Swin-Transformer
* Remove `indexing` arg
* Support dynamic inputs for t2t_vit
* Add copyright
* Fix bugs in swin transformer
* Add `pad_small_maps` option
* Update swin transformer
* Handle `attn_mask` in checkpoints of swin
* Imporve by comments
2022-03-03 13:10:12 +08:00
24ae53a4a0
[Enhance] Return code 1 if scripts runs failed. ( #694 )
...
* return code 1 when run failed
* return code 1 when run failed
2022-03-03 10:54:16 +08:00
dd45491d14
[Docs] Add MMRotate in README. ( #717 )
2022-03-02 18:57:12 +08:00
9fd35dd7b5
[Reproduction] Reproduce training results of DeiT. ( #711 )
...
* Update deit training settings
* Update decay config
* Add mixup&cutmix and drop path rate
* Update training configs
* Update model-zoo
* Add comments
2022-03-02 14:23:10 +08:00
1a28f9ace6
[Feature] Add `evaluate` function for ConcatDataset. ( #650 )
...
* Add `evaluate` function for ConcatDataset
* Remove newline in log.
* Fix lint
* Specify mmcv version in Windows CI
2022-02-28 12:46:17 +08:00
1214df083d
[Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. ( #696 )
...
* some change to mmcls/models/losses/utils.py:convert_to_one_hot()
* fixed problem: line too long
* fixed wrong output shape
* fixed lint PEP8 E128
* fix lint
* fix lint
* add unit tests
Co-authored-by: Ezra-Yu <1105212286@qq.com>
2022-02-28 10:20:31 +08:00
5f7322c211
[Fix] Fix Conformer forward with irregular input size. ( #686 )
...
* Auto pad the feature map to be divisible by 4.
* Add unit tests
* Use `trans_down_stride` instead of 4.
2022-02-28 10:17:52 +08:00
28e3bc8fd9
[CI] Add Windows CI. ( #708 )
...
* Update workflow to add Windows CI
* Add some optional requirements to requirements/optional.txt
* Update setup.py
* Update PyTorch version in Windows CI
* Update CI and requirement
* Replace `tempfile.NamedTemporaryFile` to avoid unit test error on
Windows
* Update tests
* Update CI
* Add OpenCV installation
* Update CI
2022-02-28 10:06:08 +08:00
1b5712a551
[Enhance] Add a new pre-commit-hook to automatically add a copyright. ( #710 )
2022-02-25 11:48:07 +08:00
8374c05b0a
[Enhance] Add ConvNeXt pretrain models on ImageNet-1k. ( #707 )
...
* [Docs] Add ConvNeXt pretrain models on ImageNet-1k
* Update README
* Update metafile
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-02-24 16:37:59 +08:00
9ee41577d0
[Enhance] Add deprecation message for deploy tools. ( #697 )
...
* add derepcation message for deploy tools
* add space
* Update tools/deployment/onnx2tensorrt.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* remove colorama package
* use plaintext to avoid import mmcv
Co-authored-by: Ma Zerun <mzr1996@163.com>
2022-02-23 22:24:44 +08:00
7fcaedcbfb
[Feature] Add ResNetV1c. ( #692 )
...
* add ResNetV1c
* add unit tests
* fix lint
* update docstring
* fix lint
2022-02-23 11:36:33 +08:00
43024cda73
[Feature] Support gem pooling ( #677 )
...
* add gem pooling
* add example config
* fix params
* add assert
* add param clamp
* add test assert
* add clamp
* fix conflict
2022-02-17 02:17:36 +08:00
fcd57913ae
[Enhance] Upgrade isort pre-commit hooks. ( #687 )
...
* Use new version flake8 and isort hooks
* Fix missing copyright
2022-02-17 02:17:20 +08:00
44e9902979
[Refactor] Use `--gpu-id` instead of `--gpu-ids` in non-distributed multi-gpu training/testing. ( #688 )
2022-02-17 02:17:01 +08:00
bca695b684
[Refactor] Remove deprecation. ( #633 )
...
* Remove deprecated `--options` in some tools
* Remove deprecated eval hooks and fp16 hooks
* Remove deprecated mixup&cutmix, pretrained and return_tuple arguments.
* Remove deprecated `deprecated_options` in multi label dataset
* Remove deprecated `formating.py`
* Remove deprecated default mode of `LabelSmoothLoss`
* Revert pretrained argument, and remove it's deprecation sign.
* Update unit test
2022-02-10 09:30:18 +08:00
takuoko
Ma Zerun
Weihao Yu
Yuan Liu
XiaobingZhang
Ezra-Yu
Hubert
huyu
Hydrion-Qlz
Ming-Hsuan-Tu
kusakana
HumberMe
RunningLeon