060b0ed3b5
[Feature] Support ConvNeXt-V2 backbone. ( #1294 )
...
* [Feature] Support ConvNeXt-V2.
* Use registry of mmcls instead of mmengine.
* Add README.
* Add unit tests and docs.
2023-01-06 16:13:41 +08:00
bac181f393
[Feature] Support Multi-task. ( #1229 )
...
* unit test for multi_task_head
* [Feature] MultiTaskHead (#628 , #481 )
* [Fix] lint for multi_task_head
* [Feature] Add `MultiTaskDataset` to support multi-task training.
* Update MultiTaskClsHead
* Update docs
* [CI] Add test mim CI. (#879 )
* [Fix] Remove duplicated wide-resnet metafile.
* [Feature] Support MPS device. (#894 )
* [Feature] Support MPS device.
* Add `auto_select_device`
* Add unit tests
* [Fix] Fix Albu crash bug. (#918 )
* Fix albu BUG: using albu will cause the label from array(x) to array([x]) and crash the trainning
* Fix common
* Using copy incase potential bug in multi-label tasks
* Improve coding
* Improve code logic
* Add unit test
* Fix typo
* Fix yapf
* Bump version to 0.23.2. (#937 )
* [Improve] Use `forward_dummy` to calculate FLOPS. (#953 )
* Update README
* [Docs] Fix typo for wrong reference. (#1036 )
* [Doc] Fix typo in tutorial 2 (#1043 )
* [Docs] Fix a typo in ImageClassifier (#1050 )
* add mask to loss
* add another pipeline
* adpat the pipeline if there is no mask
* switch mask and task
* first version of multi data smaple
* fix problem with attribut by getattr
* rm img_label suffix, fix 'LabelData' object has no attribute 'gt_label'
* training without evaluation
* first version work
* add others metrics
* delete evaluation from dataset
* fix linter
* fix linter
* multi metrics
* first version of test
* change evaluate metric
* Update tests/test_models/test_heads.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_models/test_heads.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* add tests
* add test for multidatasample
* create a generic test
* create a generic test
* create a generic test
* change multi data sample
* correct test
* test
* add new test
* add test for dataset
* correct test
* correct test
* correct test
* correct test
* fix : #5
* run yapf
* fix linter
* fix linter
* fix linter
* fix isort
* fix isort
* fix docformmater
* fix docformmater
* fix linter
* fix linter
* fix data sample
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update data sample
* update head
* update head
* update multi data sample
* fix linter
* fix linter
* fix linter
* fix linter
* fix linter
* fix linter
* update head
* fix problem we don't set pred or gt
* fix problem we don't set pred or gt
* fix problem we don't set pred or gt
* fix linter
* fix : #2
* fix : linter
* update multi head
* fix linter
* fix linter
* update data sample
* update data sample
* fix ; linter
* update test
* test pipeline
* update pipeline
* update test
* update dataset
* update dataset
* fix linter
* fix linter
* update formatting
* add test for multi-task-eval
* update formatting
* fix linter
* update test
* update
* add test
* update metrics
* update metrics
* add doc for functions
* fix linter
* training for multitask 1.x
* fix linter
* run flake8
* run linter
* update test
* add mask in evaluation
* update metric doc
* update metric doc
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update metric doc
* update metric doc
* Fix cannot import name MultiTaskDataSample
* fix test_datasets
* fix test_datasets
* fix linter
* add an example of multitask
* change name of configs dataset
* Refactor the multi-task support
* correct test and metric
* add test to multidatasample
* add test to multidatasample
* correct test
* correct metrics and clshead
* Update mmcls/models/heads/cls_head.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update cls_head.py documentation
* lint
* lint
* fix: lint
* fix linter
* add eval mask
* fix documentation
* fix: single_label.py back to 1.x
* Update mmcls/models/heads/multi_task_head.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* Remove multi-task configs.
Co-authored-by: mzr1996 <mzr1996@163.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Ming-Hsuan-Tu <alec.tu@acer.com>
Co-authored-by: Lei Lei <18294546+Crescent-Saturn@users.noreply.github.com>
Co-authored-by: WRH <12756472+wangruohui@users.noreply.github.com>
Co-authored-by: marouaneamz <maroineamil99@gmail.com>
Co-authored-by: marouane amzil <53240092+marouaneamz@users.noreply.github.com>
2022-12-30 10:36:00 +08:00
5547f4cac4
[Feature] Add TinyViT for dev-1.x. ( #1042 )
...
* [Feature] add TinyViT for dev-1.x
* [Feature] update readme
* fix lint error
* refactor the code
* [Update] update the args
* [Update] add unit test and fix bugs
* Rename the configuration file
* delete invalid files
* [Feature] update tinyvit readme
* [Feature] update tinyvit readme
* [Feature] update metafile
* Update tinyvit metafile
2022-12-20 13:04:00 +08:00
2535c1ecd7
[Feature] Support EVA. ( #1239 )
...
* add eva
* add eva
* add eva
* sklearn -> scikit-learn
* add large
* Update model names and links.
* Fix resize pos embed error when loading fp16 weight.
* Remove verbose configs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-14 13:21:33 +08:00
d80ec5a4b8
[Refactor] Refactor BEiT backbone and support v1/v2 inference. ( #1144 )
...
* refactor beit backbone
* use LinearClsHead
* fix mean and std value
* fix lint
* support inference if beit-v2
* update encoder layer and init
* update
* add ut
* add prepare_relative_position_bias_table function
* add cls_token
* fix lint
* add pos_embed check
* update metafile and readme
* update weights link
* update link of weights
* update metafile
* update
* update docstrings
* update according to review
* rename readme
* update docstring
* fix lint
2022-11-29 12:56:33 +08:00
b526f018db
[Fix] Fix broken inference api because of the modification in data
...
preprocessor.
2022-10-18 17:50:35 +08:00
29f066f7fb
[Improve] Speed up data preprocessor. ( #1064 )
...
* [Improve] Speed up data preprocessor.
* Add ClsDataSample serialization override functions.
* Add unit tests
* Modify configs to fit new mixup args.
* Fix `num_classes` of the ImageNet-21k config.
* Update docs.
2022-10-17 17:08:18 +08:00
a49c3076e1
[Feature] Support DeiT3. ( #1065 )
...
* deit3
deit3
lint
* add tools and test
* deit3
* deit3
* fix preprocess
* lint
* Update config names and checkpoint paths
* Update convert tools to use mmengine, and fix docstring.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-10 14:54:20 +08:00
81f10ccbb7
[Feature] Support EdgeNeXt for dev-1.x. ( #1037 )
...
* [Feature] add EdgeNeXt for dev-1.x
* [Feature] add update
* Imporve EdgeNeXt
* Fix div error in low pytorch version
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-21 13:27:04 +08:00
e4e8047563
[Feature] Support Swin Transformer V2 for MMCLS 1.x. ( #1029 )
...
* [Feature] support swin v2 in mmcls 1.x
* fix configs
* fix clamp max
* Fix meshgrid
* Imporve swin v2
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 15:45:27 +08:00
aaf127c5e1
[Feature] Add efficientformer Backbone for MMCls 1.x. ( #1031 )
...
* rebase
* update filename
* update URL
* update UT
* fix lint
* update head
* add efficientformer
* update filename
* update UT
* fix lint
* update configs
* rebase
* fix unit tests
* Fix comments and docs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 14:56:45 +08:00
0b61b76325
[Docs] Add Migration documentation. ( #997 )
...
* [Docs] Add Migration documentation.
* Fix the train_cfg argument
* [WIP] Improve according to comments
* Imporve according to comments.
2022-08-31 13:31:48 +08:00
2b88df4484
[Refactor] Refactor the data flow. ( #989 )
...
* [Refactor] Refactor the data flow.
* Add comments about data preprocessor.
* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
b7d0d521eb
[Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. ( #986 )
...
* [Refactor] Refactor the rest data transforms and fix some docstring.
* Use `mmengine` instead of `mmcv`
* Refactor KFold dataset tools
* Fix docstring according to comments
2022-08-24 15:59:02 +08:00
e4252d6848
[Feature]: Support BEiT Transformer layer. ( #919 )
...
* [Feature]: Add BEiT-style transformer encoder layer
* [Feature]: Add average token
* [Fix]: Fix lint
* [Fix]: Refactor CAE config
* [Fix]: Change cv2 backend to pillow backend
* [Fix]: Fix MAE and CAE reshape bug
* [Feature]: Add freeze vit layers
* [Feature]: Add mc
* [Fix]: Fix lint
* [Fix]: Fix dataset bug
* [Fix]: Delete cae selfsup config
* [Fix]: docstring
* [Refactor]: Add init_values to layer_scalue_init_value
* [Fix]: Refine the docstring of avg_token
* [Fix]: Call super init weight in beit attention
* [Fix]: remove mc
* [Fix]: Fix docstring
* [Fix]: Fix docstring
* [Fix]: Fix lint
* [Fix]: Fix init_value bug and change the logic of outputting cls token
* [Fix]: Fix docstring
2022-08-17 00:07:06 +08:00
43e60ad5a6
[Improve] Rename `mmcls.data` to `mmcls.structures`. ( #941 )
2022-07-29 14:18:13 +08:00
6ed3860ea6
refactor UTs
2022-07-18 11:11:13 +08:00
0f30c392a3
from core to engine
2022-07-18 11:11:13 +08:00
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
daa716b112
Add some type hints
2022-07-18 11:11:13 +08:00
69e5ab065e
Change "Default to" to "Defaults to"
2022-07-18 11:11:13 +08:00
375fe68f12
Imporve according to comments
2022-07-18 11:11:13 +08:00
f3299b4ca2
[Refactor] Refactor batch augmentations
2022-07-18 11:11:13 +08:00
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
088d5b5add
[Refactor] Inherit all registries from MMEngine and use inherited registries.
2022-07-18 10:53:27 +08:00
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +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
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
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
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
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
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
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
d29037e8d1
[Enhance] Suport Mixup&Cutmix for multi-label task.
2022-01-21 11:30:58 +08:00
7dfc9e4a85
Bump version to v0.19.0 ( #623 )
2021-12-31 12:55:47 +08:00
0bbbb04429
[Enhancement] Improve the accuracy of ResNet ( #572 )
...
* add itp timm
* minor update
* minor update
* minor update
* add rep aug, minor update on configs
* minor update
* add target threshold
* add decaymulti
* minor update
* minor update
* add lbl smooth
* update lr
* reorganize config files and code
* minor bugfixes
* remove unused parts and minor fixes on cfg
* critical bugfix, add test and cfg update
* refactor code
* update doc string
* remove duplicate code
* refactor drop path in resnet
* rename
* Modify configs and add README&metafile
* Update metafile
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-29 10:58:42 +08:00
2932f9d8a3
[Refactor] Refator ViT (Continue #295 ) ( #395 )
...
* [Squash] Refator ViT (from #295 )
* Use base variable to simplify auto_aug setting
* Use common PatchEmbed, remove HybridEmbed and refactor ViT init
structure.
* Add `output_cls_token` option and change the output format of ViT and
input format of ViT head.
* Update unit tests and add test for `output_cls_token`.
* Support out_indices.
* Standardize config files
* Support resize position embedding.
* Add readme file of vit
* Rename config file
* Improve docs about ViT.
* Update docstring
* Use local version `MultiheadAttention` instead of mmcv version.
* Fix MultiheadAttention
* Support `qk_scale` argument in `MultiheadAttention`
* Improve docs and change `layer_cfg` to `layer_cfgs` and support
sequence.
* Use init_cfg to init Linear layer in VisionTransformerHead
* update metafile
* Update checkpoints and configs
* Imporve docstring.
* Update README
* Revert GAP modification.
2021-10-18 16:07:00 +08:00
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
0184527bd4
[Enhance] Enhence SE layer to support custom squeeze channels. ( #417 )
...
* add enhenced SE
* Update
* rm basechannel
* fix docstring
* Update se_layer.py
fix docstring
2021-08-20 13:31:44 +08:00
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
33a4c0fe9f
[Fix] Use local ATTENTION registery to avoid conflict with other repo. ( #375 )
2021-07-27 17:59:05 +08:00
d04ebc1eb5
[Docs] Add API Reference in the docs ( #342 )
...
* Add API inference in the docs and fix readthedocs config.
* Replace some relative link in docs.
* Format docstring for reStructuredText syntax.
* Fix vit paper link
* Fix docstring of `show_results` function in `BaseClassifier`.
2021-07-14 15:06:50 +08:00
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
71621a5f62
Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 ( #347 )
2021-07-07 11:55:53 +08:00
076ee10cac
[Feature] Add swin-transformer model. ( #271 )
...
* Add swin transformer archs S, B and L.
* Add SwinTransformer configs
* Add train config files of swin.
* Align init method with original code
* Use nn.Unfold to merge patch
* Change all ConfigDict to dict
* Add init_cfg for all subclasses of BaseModule.
* Use mmcv version init function
* Add Swin README
* Use safer cfg copy method
* Improve docstring and variable name.
* Fix some difference in randaug
Fix BGR bug, align scheduler config.
Fix label smoothing parameter difference.
* Fix missing droppath in attn
* Fix bug of relative posititon table if window width is not equal to
height.
* Make `PatchMerging` more general, support kernel, stride, padding and
dilation.
* Rename `residual` to `identity` in attention and FFN.
* Add `auto_pad` option to auto pad feature map
* Improve docstring.
* Fix bug in ShiftWMSA padding.
* Remove unused `key` and `value` in ShiftWMSA
* Move `PatchMerging` into utils and use common `PatchEmbed`.
* Use latest `LinearClsHead`, train augments and label smooth settings.
And remove original `SwinLinearClsHead`.
* Mark some configs as "Evalution Only".
* Remove useless comment in config
* 1. Move ShiftWindowMSA and WindowMSA to `utils/attention.py`
2. Add docstrings of each module.
3. Fix some variables' names.
4. Other small improvement.
* Add unit tests of swin-transformer and patchmerging.
* Fix some bugs in unit tests.
* Fix bug of rel_position_index if window is not square.
* Make WindowMSA implicit, and add unit tests.
* Add metafile.yml, update readme and model_zoo.
2021-07-01 09:30:42 +08:00
bee0ac6b56
[Refactor]Modify patchembed ( #330 )
...
* add mytrain.py for test
* test before layers
* test attr in layers
* test classifier
* delete mytrain.py
* add patchembed and hybridembed
* add patchembed and hybridembed to __init__
* test patchembed and hybridembed
* fix some comments
2021-06-30 20:48:04 +08:00
65410b05ad
Fix Mobilenetv3 structure and add pretrained model ( #291 )
...
* Refactor Mobilenetv3 structure and add ConvClsHead.
* Change model's name from 'MobileNetv3' to 'MobileNetV3'
* Modify configs for MobileNetV3 on CIFAR10.
And add MobileNetV3 configs for imagenet
* Fix activate setting bugs in MobileNetV3.
And remove bias in SELayer.
* Modify unittest
* Remove useless config and file.
* Fix mobilenetv3-large arch setting
* Add dropout option in ConvClsHead
* Fix MobilenetV3 structure according to torchvision version.
1. Remove with_expand_conv option in InvertedResidual, it should be decided by channels.
2. Revert activation function, should before SE layer.
* Format code.
* Rename MobilenetV3 arch "big" to "large".
* Add mobilenetv3_small torchvision training recipe
* Modify default `out_indices` of MobilenetV3, now it will change
according to `arch` if not specified.
* Add MobilenetV3 large config.
* Add mobilenetv3 README
* Modify InvertedResidual unit test.
* Refactor ConvClsHead to StackedLinearClsHead, and add unit tests.
* Add unit test for `simple_test` of `StackedLinearClsHead`.
* Fix typo
Co-authored-by: Yidi Shao <ydshao@smail.nju.edu.cn>
2021-06-27 23:19:36 +08:00
3a08db9182
[Feature]Add augments to models/utils ( #278 )
...
* add mytrain.py for test
* test before layers
* test attr in layers
* test classifier
* delete mytrain.py
* add rand_bbox_minmax rand_bbox and cutmix_bbox_and_lam to BaseCutMixLayer
* add mixup_prob to BatchMixupLayer
* add cutmixup
* add cutmixup to __init__
* test classifier with cutmixup
* delete some comments
* set mixup_prob default to 1.0
* add cutmixup to classifier
* use cutmixup
* use cutmixup
* fix bugs
* test cutmixup
* move mixup and cutmix to augment
* inherit from BaseAugment
* add BaseAugment
* inherit from BaseAugment
* rename identity.py
* add @
* build augment
* register module
* rename to augment.py
* delete cutmixup.py
* do not inherit from BaseAugment
* add augments
* use augments in classifier
* prob default to 1.0
* add comments
* use augments
* use augments
* assert sum of augmentation probabilities should equal to 1
* augmentation probabilities equal to 1
* calculate Identity prob
* replace xxx with self.xxx
* add comments
* sync with augments
* for BC-breaking
* delete useless comments in mixup.py
2021-06-20 09:44:51 +08:00
4ca21c7d03
[WIP] Refactoring weights initialization ( #270 )
...
* [WIP] Refactoring weights initialization
* fix lint and constant init cfg
* fix pretrained bug
* fix typo
* fix isort
* revise model utils
2021-06-10 10:54:34 +08:00
Ma Zerun
Colle
Songyang Zhang
takuoko
Yixiao Fang
Hubert
Ezra-Yu
Yuan Liu
Ezra-Yu
XiaobingZhang
Zhicheng Chen
whcao
Miao Zheng