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
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
kusakana
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
Ma Zerun
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
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
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
Zhicheng Chen
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
Ezra-Yu
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
Ma Zerun
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
Ma Zerun
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
Ma Zerun
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
Ma Zerun
6f25bebe42
[Docs] Update docs. ( #600 )
...
* [Docs] The template provides united Docs and OpenMMLab drop-lists in
banner. Remove them from our `conf.py`.
* Move docs of different languages to the same folder
2021-12-15 10:34:34 +08:00