4d4b22377d
Add browse dataset
2022-07-18 10:53:56 +08:00
65f3b2221d
Modify config relates to logger and checkpoint
2022-07-18 10:53:28 +08:00
58d9f649ed
Refactor scheduler configuration
2022-07-18 10:53:28 +08:00
2f2aa3037c
Refactor default hooks configs
2022-07-18 10:53:28 +08:00
93a27c8324
[Feature] Add `PackClsInputs` and use `LoadImageFromFile`, `Resize` & `RandomFlip` in MMCV.
2022-07-18 10:53:28 +08:00
58b21ee56f
[Model] Add IPU ViT model
2022-06-03 18:15:28 +08:00
824fbcbaae
[Refactor] Use mdformat instead of markdownlint to format markdown. ( #844 )
...
* [Refactor] Use mdformat instead of markdownlint to format markdown.
* Update unavailiable api links in tutorials
* Update CONTRIBUTING.md
* Use mdformat==0.7.9 to support Python 3.6
2022-06-02 15:22:01 +08:00
7c5ddb1e5b
Bump version to v0.23.0 ( #809 )
2022-05-01 21:58:33 +08:00
1d6fbe0efe
[Fix] Fix lint and mmcv version requirement for IPU.
2022-04-29 22:33:29 +08:00
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
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
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
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
f0ee5dcb2a
Update checkpoint path of ResNet-50 on CUB dataset
2022-03-31 10:23:33 +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
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
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
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
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
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
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
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
a7f8e96b31
Bump version to v0.20.1 ( #685 )
2022-02-07 11:46:50 +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
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
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
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
cdf569a805
Fix ResNet rsb config
2021-12-31 11:12:21 +08:00
e1a4020168
Fix ResNet rsb config
2021-12-31 11:08:11 +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
e7c06b8541
[Reproduction] Reproduce RegNetX training accuracy. ( #587 )
...
* add preciseBN hook
* add precise-bn config
* add proority of preciseBN
* Add precise BN hook
* Update PreciseBN hook
* add config example & pre-commit
* Update PBN
* Add unit test
* Update unit tests
* mv preciseBN to mmcls.runner
* rm extra sleep()
* merge latest
* update code
* fix lint
* update unit tests
* add regnetx configs
* update prcise BN in configs
* fix lint
* add checkpoints and logs
* fix conflict
* fix lint
* update model_zoo and model-index
* remove some changes
* update readme.md
* remove precise_bn.py
2021-12-28 17:34:27 +08:00
159b38d276
[Reproduction] Reproduce training results of T2T-ViT ( #610 )
...
* Add cosine cool down lr updater
* Use ema hook
* Update decay mult
* Update configs.
* Update T2T-ViT readme and format all readme
* Update swin readme
* Update tnt readme
* Add docstring for `CosineAnnealingCooldownLrUpdaterHook`.
* Update t2t readme and metafile
2021-12-28 15:09:40 +08:00
d1d9ebfc67
[Feature] Add Precise BN ( #401 )
...
* add preciseBN hook
* add precise-bn config
* add proority of preciseBN
* Add precise BN hook
* Update PreciseBN hook
* add config example & pre-commit
* Update PBN
* Add unit test
* Update unit tests
* mv preciseBN to mmcls.runner
* rm extra sleep()
* merge latest
* update code
* fix lint
* update unit tests
* update unit tests and add warning
* update docstring and renamme filename
* remove model warpper
* add copyright
* Imporve docs and update some variable names
* Update resnet50_8xb32-coslr-preciseBN_in1k.py
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-24 16:28:42 +08:00
131d8c6296
[Feature] Add CAM visualization tool ( #577 )
...
* add cam-grad tool
* refactor cam-grad tool
* add docs
* update docs
* Update docs and support Transformer
* remove pictures and use link
* replace example img and finish EN docs
* improve docs
* improve code
* Fix MobileNet V3 configs
* Refactor to support more powerful feature extraction.
* Add unit tests
* Fix unit test
* fix distortion of visualization exapmles in docs
* fix distortion
* fix distortion
* fix distortion
* merge master
* merge fix conficts
* Imporve the tool
* Support use both attribute name and index to get layer
* add default get_target-layers
* add default get_target-layers
* update docs
* update docs
* add additional printt info when not using target-layers
* Imporve docs
* Fix enumerate list.
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-23 18:53:40 +08:00
643fb192cd
[Enhance] Enhance feature extraction function. ( #593 )
...
* Fix MobileNet V3 configs
* Refactor to support more powerful feature extraction.
* Add unit tests
* Fix unit test
* Imporve according to comments
* Update checkpoints path
* Fix unit tests
* Add docstring of `simple_test`
* Add docstring of `extract_feat`
* Update model zoo
2021-12-17 15:55:02 +08:00
f9a2b04cee
[Feature] Add DeiT backbone and checkpoints. ( #576 )
...
* Support DeiT backbone.
* Use hook to automatically resize pos embed
* Update ViT training setting
* Add deit configs and update docs
* Fix vit arch assertion
* Remove useless init function
* Add unit tests.
* Fix resize_pos_embed for DeiT
* Improve according to comments.
2021-12-15 22:44:57 +08:00
18f6bb0b10
[Feature] Implement the conformer backbone. ( #494 )
...
* implement the conformer
* format code style
* format code style
* reuse the TransformerEncoderLayer in the vision_transformer.py
* Modify variable name
* delete unused params
* Remove warning info in Conformer head since it already exists in
Conformer.
* Rename some variables
* Add unit tests
* Use `getattr` instead of `get_submodule`.
* Remove some useless layers
* Refactor conformer and add configs
* Update configs and add metafile.
* Fix unit tests
* Update README
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 14:00:17 +08:00
fc8adbc149
[Enhance] Add metafile, readme and converted models for MLP-Mixer ( #539 )
...
* add pth converter
* minor update on config files, add metafile and readme
* add missing readme and minor fixes
* minor fixes
* Update config names and checkpoint download link
* Update model_zoo.md
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-24 19:04:19 +08:00
f3fbc8b90b
[Docs] Add abstract and image for every paper. ( #546 )
2021-11-24 17:23:37 +08:00
7977dc8e2d
[Improvement] Rename config files according to the config name standard. ( #508 )
...
* Update tnt config
* Rename config files
* Update docs
* Update metafile in dev_scripts
* Fix some files
* Remove fp16 metafile and README.
* Fix names in metafiles.
2021-11-19 14:20:35 +08:00
3eafc5cbe1
[Feature] Add MLP Mixer Backbone. ( #528 )
...
* Add Mlp Mixer
* Fix token_mix FNN add_identity and add out_indices
* Fix mlp_mixer unit test
2021-11-15 18:25:22 +08:00
3f52488d52
[Docs] Add the top-5 acc in ResNet-CIFAR README. ( #531 )
...
* Update README.md
* Update README.md
2021-11-12 16:13:41 +08:00
d1473e4a7f
[Dependency] Update mmcv dependency version ( #509 )
...
* Update mmcv dependency version
* Add code info in some metafiles
2021-11-02 18:08:30 +08:00
b8c3825bb3
Fix uppercase in repvgg metafile
2021-11-02 15:13:26 +08:00
5ac0a1561d
[Docs] Fix missing checkpoints links in T2T-ViT.
2021-10-29 10:42:10 +08:00
fffa30dd48
[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. ( #467 )
...
* add t2t backbone
* register t2t_vit
* add t2t_vit config
* [Temp] Align posterize transform with timm.
* Fix lint
* Refactor t2t-vit
* Add config for t2t-vit
* Add metafile and README for t2t-vit
* Add unit tests
* configs
* Update metafile and README
* Improve docstring
* Fix batch size which should be 8x64 instead of 8x128
* Fix typo
* Update model zoo
* Update training augments config.
* Move some arguments of T2TModule to T2TViT
* Update docs.
* Update unit test
Co-authored-by: HIT-cwh <2892770585@qq.com>
2021-10-29 10:37:16 +08:00
Ezra-Yu
yingfhu
mzr1996
Hu Di
Hubert
takuoko
Weihao Yu
kusakana
Zhicheng Chen
Zhiliang Peng
xcnick
huoshuai-dot