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
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
a7f8e96b31
Bump version to v0.20.1 ( #685 )
2022-02-07 11:46:50 +08:00
e0edffb353
Merge branch 'dev'
2022-01-31 12:00:17 +08:00
fb6d6f18d9
Bump version to v0.20.0 ( #679 )
...
* Bump version to v0.20.0
* Improve according to comments
2022-01-31 11:51:49 +08:00
8ef5aeda40
[Enhance] New-style CPU training and inference. ( #674 )
...
* [Enhance] New-style CPU training and inference.
* Add version check in CPU training/test
2022-01-30 20:49:54 +08:00
5de480ea9e
[Feature] Support HRNet and add pre-trained models ( #660 )
...
* Support HRNet
* Add HRNet configs
* Fix a bug in backward
* Add configs and update docs.
* Not use bias in conv before batch norm
* Defaults to use `norm_eval=False`
* Add unit tests and support out_channels in HRFuseScales
* Update checkpoint path
* Update docstring.
* Remove incorrect files
* Improve according to comments
2022-01-28 10:54:14 +08:00
dc456a0c2c
[Feature] Support ConvNeXt ( #670 )
...
* Support ConvNeXt
* Add configs of ConvNeXt
* Update dev scripts
* Update docs.
* Use new style README
* Add unit tests.
* Update README
* Imporve according to comments
* Modify refers to timm.
* Imporve according to comments
2022-01-28 10:36:45 +08:00
8488a784f0
[Fix] Fix potential unexcepted behaviors if `metric_options` is not specified in multi-label evaluation. ( #647 )
...
* Update multi_label.py
* Fix lint
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-27 10:27:38 +08:00
94eb3ff83e
[Feature] Add Twins backbone and convert checkpoints. ( #642 )
...
* add twins backbone
* add position_encoding
* refactor twins
* Supplemental unit tests
* update docstring and readme
* update docstring and readme
* update docstring and readme
* update docstring
* update docstring
* update docstring
* update docstring
* remove note
* update doc and docstring
* update docstring
* update docstring
* use abstract pdf link and rename yamlfile
* Update model link
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-27 10:25:05 +08:00
f552419e45
[Enhance] Add setup multi-processing both in train and test. ( #671 )
2022-01-27 10:18:36 +08:00
833152b1f4
[Docs] Update README in configs according to OpenMMLab standard. ( #672 )
...
* Update README according to OpenMMLab standard.
* Update model zoo docs generation.
* Revert modification for paperlink
2022-01-26 18:26:01 +08:00
a451d63d0e
Avoid CI for PRs that only modifiy configs
2022-01-25 17:52:37 +08:00
bd397f790f
[Enhance] Rewrite channel split operation in ShufflenetV2 ( #632 )
...
* replace chunk op
* shufflenetv2 config
2022-01-25 14:45:28 +08:00
b5bd87d7fa
[Docs] Fix some typo in docs.
2022-01-25 12:32:06 +08:00
a865d64208
[Refactor] Deprecate the support for "python setup.py test" ( #646 )
...
* Deprecate the support for "python setup.py test"
* Update requirements
2022-01-25 12:15:12 +08:00
d56170a734
[Feature] Support EfficientNet ( #649 )
...
* add config for resnest test
* fix config
* add label smoothing
* add memcached
* minor fix
* fix bug
* fix config
* add config
* minor fix
* fix configs
* use EResize
* change interpolation
* add more configs
* add docsting
* add unittest
* remove unnecessary changes
* minor fix
* add more docstring
* fix linting
* add efficient backbone
* add config
* add Edge Residual
* fix bug
* remove unnecessary files
* refactor
* add resize in crop to ensure crop size is output size
* fix bug and add comments
* test
* fix
* add more configs
* add more configs
* add more configs
* fix bug
* add model zoo
* fix
* reorganize code
* add edge tpu
* add edge tpu converter
* rename
* update readme
* reorganize code and config
* Rename configs of EfficientNet, and add metafile & model_zoo
* Remove `backend='pillow'`
* Add comments about EfficientNet-EdgeTPU
* Rename the convert tool of EfficientNet.
* Refactor EfficientNet and update docstring.
* Update EfficientNet-EdgeTPU config
* Fix unit tests
Co-authored-by: lixinran <lixr423@outlook.com>
Co-authored-by: lixinran <lixinran@sensetime.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 12:14:17 +08:00
Ma Zerun
Ezra-Yu
Hubert
huyu
takuoko
Ming-Hsuan-Tu
kusakana
HumberMe
RunningLeon
xiatao
WINDSKY45
Zhicheng Chen