mmpretrain/mmcls/models/heads
Ezra-Yu b0007812d6
[Enhance] Enhance ArcFaceClsHead. (#1181)
* update arcface

* fix unit tests

* add adv-margins

add adv-margins

update arcface

* rebase

* update doc and fix ut

* rebase

* update code

* rebase

* use label data

* update set-margins

* Modify Arcface related method names.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 18:10:39 +08:00
..
__init__.py [Enhance] Enhance ArcFaceClsHead. (#1181) 2022-11-21 18:10:39 +08:00
base_head.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
cls_head.py [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +08:00
conformer_head.py [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +08:00
deit_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
efficientformer_head.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
linear_head.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
margin_head.py [Enhance] Enhance ArcFaceClsHead. (#1181) 2022-11-21 18:10:39 +08:00
multi_label_cls_head.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
multi_label_csra_head.py [Feature] Migrate CSRA head to 1.x. (#1177) 2022-11-21 10:39:16 +08:00
multi_label_linear_head.py Add multi_label heads 2022-07-18 11:11:13 +08:00
stacked_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
vision_transformer_head.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00