mmpretrain/mmcls
zzc98 9506241f73
[Feature] Add arcface head. (#926)
* feat: add arcface head

* feat: add arcface head

* update arcface

* refactor archface head

* update archface head

* update archface head

* use torch.cat instead of torch.hstack to fix ci

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-02 17:45:33 +08:00
..
apis [Fix] Fix broken inference api because of the modification in data 2022-10-18 17:50:35 +08:00
datasets [Docs] Fix some docstrings. 2022-10-10 11:17:48 +08:00
engine [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00
evaluation [Docs] Add custom evaluation docs (#1130) 2022-11-01 18:54:06 +08:00
models [Feature] Add arcface head. (#926) 2022-11-02 17:45:33 +08:00
structures [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
utils [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00
visualization [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
__init__.py Update requirements. (#1083) 2022-10-12 10:57:37 +08:00
registry.py
version.py Bump version to v1.0.0rc2. (#1084) 2022-10-12 16:52:31 +08:00