mirrors/mmpretrain
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 14:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity
653 Commits 8 Branches 41 Tags
Commit Graph

220 Commits

Author SHA1 Message Date
Ma Zerun
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
Ma Zerun
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
Ma Zerun
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
Yixiao Fang
5665b8349a
[Refactor] Refactor import dependency. (#977) 
Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-22 15:17:33 +08:00
Ma Zerun
20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. (#970) 
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
 2022-08-22 15:02:08 +08:00
Yuan Liu
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
Ma Zerun
43e60ad5a6
[Improve] Rename mmcls.data to mmcls.structures. (#941) 2022-07-29 14:18:13 +08:00
Ezra-Yu
9118b97239 refactor evaluation 2022-07-18 11:11:13 +08:00
Ezra-Yu
6ed3860ea6 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu
0f30c392a3 from core to engine 2022-07-18 11:11:13 +08:00
mzr1996
24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
Ezra-Yu
e9342d9e4c Add multi_label heads 2022-07-18 11:11:13 +08:00
yingfhu
62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 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
375fe68f12 Imporve according to comments 2022-07-18 11:11:13 +08:00
mzr1996
f3299b4ca2 [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
mzr1996
cecff79a79 Rename 'feat' mode to 'tensor' mode 2022-07-18 11:11:13 +08:00
mzr1996
ee6e585e41 Mazerun/base classifier 2022-07-18 11:11:13 +08:00
mzr1996
0537c4d70c [Feature] Add ClsVisualizer. 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
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
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
Weihao Yu
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
Yuan Liu
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
XiaobingZhang
875195efe9
[Enhance] Improve CPE performance by reduce memory copy. (#762) 2022-04-02 11:12:20 +08:00
Ezra-Yu
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
Ezra-Yu
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
Ma Zerun
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
Hubert
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
takuoko
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
Ma Zerun
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
HumberMe
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
Ma Zerun
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
Ezra-Yu
7fcaedcbfb
[Feature] Add ResNetV1c. (#692) 
* add ResNetV1c

* add unit tests

* fix lint

* update docstring

* fix lint
 2022-02-23 11:36:33 +08:00
takuoko
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
Ma Zerun
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
Ma Zerun
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
Ma Zerun
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
Ma Zerun
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
Ezra-Yu
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
Ezra-Yu
bd397f790f
[Enhance] Rewrite channel split operation in ShufflenetV2 (#632) 
* replace chunk op

* shufflenetv2 config
 2022-01-25 14:45:28 +08:00
Zhicheng Chen
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
Yosuke Shinya
16864c7495
[Feature] Support features_only in TIMMBackbone (#668) 
* Support features_only in TIMMBackbone

based on https://github.com/open-mmlab/mmsegmentation/pull/998

* update test for mmdet

* fix unit test for build_without_timm

* Update docstring

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-01-25 11:22:23 +08:00