mmpretrain/mmcls/models/heads
Ezra-Yu 6474ea2fc0
[Feature] Support EfficientFormer. (#954)
* add efficient backbone

* Update Readme and metafile

* Add unit tests

* fix confict

* fix lint

* update efficientformer head unit tests

* update README

* fix unit test

* fix Readme

* fix example

* fix typo

* recover api modification

* Update EfficiemtFormer Backbone

* fix unit tests

* add efficientformer to readme and model zoo
2022-08-16 23:38:08 +08:00
..
__init__.py [Feature] Support EfficientFormer. (#954) 2022-08-16 23:38:08 +08:00
base_head.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
cls_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
conformer_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
deit_head.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
efficientformer_head.py [Feature] Support EfficientFormer. (#954) 2022-08-16 23:38:08 +08:00
linear_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
multi_label_csra_head.py [Feature] Support CSRA head. (#881) 2022-08-04 18:15:51 +08:00
multi_label_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
multi_label_linear_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
stacked_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
vision_transformer_head.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00