mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/mmcls/models/heads
History
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
..
__init__.py [Feature] Add DeiT backbone and checkpoints. (#576) 2021-12-15 22:44:57 +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 [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
linear_head.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +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 [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00