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/__init__.py

9 lines
272 B
Python
Raw Normal View History

Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
from .cls_head import ClsHead
from .linear_head import LinearClsHead
[Feature] Add heads and config for multilabel task (#145) * resolve conflicts add heads and config for multilabel tasks * minor change * remove evaluating mAP in head * add baseline config * add configs * reserve only one config * minor change * fix minor bug * minor change * minor change * add unittests and fix docstrings
2021-01-25 18:10:14 +08:00
from .multi_label_head import MultiLabelClsHead
from .multi_label_linear_head import MultiLabelLinearClsHead
Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
[Feature] Add heads and config for multilabel task (#145) * resolve conflicts add heads and config for multilabel tasks * minor change * remove evaluating mAP in head * add baseline config * add configs * reserve only one config * minor change * fix minor bug * minor change * minor change * add unittests and fix docstrings
2021-01-25 18:10:14 +08:00
__all__ = [
'ClsHead', 'LinearClsHead', 'MultiLabelClsHead', 'MultiLabelLinearClsHead'
]