8 Commits

Author SHA1 Message Date
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
yingfhu
62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +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
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
Zhiliang Peng
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