mmpretrain/mmcls/models/necks/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .gap import GlobalAveragePooling
from .gem import GeneralizedMeanPooling
from .hr_fuse import HRFuseScales
from .reduction import LinearReduction

__all__ = [
    'GlobalAveragePooling', 'GeneralizedMeanPooling', 'HRFuseScales',
    'LinearReduction'
]
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00			`# Copyright (c) OpenMMLab. All rights reserved.`
Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00			`from .gap import GlobalAveragePooling`
[Feature] Support gem pooling (#677) * add gem pooling * add example config * fix params * add assert * add param clamp * add test assert * add clamp * fix conflict 2022-02-17 02:17:36 +08:00			`from .gem import GeneralizedMeanPooling`
[Feature] Support HRNet and add pre-trained models (#660) * Support HRNet * Add HRNet configs * Fix a bug in backward * Add configs and update docs. * Not use bias in conv before batch norm * Defaults to use `norm_eval=False` * Add unit tests and support out_channels in HRFuseScales * Update checkpoint path * Update docstring. * Remove incorrect files * Improve according to comments 2022-01-28 10:54:14 +08:00			`from .hr_fuse import HRFuseScales`
[Feature] Add reduction for neck (#978) * feat: add reduction for neck * feat: add reduction for neck * feat: add reduction for neck * feat:add linear reduction neck * feat: add reduction neck * mod out of linearReduction as tuple * fix typo * fix unit tests * fix unit tests Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> 2022-11-04 15:44:37 +08:00			`from .reduction import LinearReduction`
Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
[Feature] Add reduction for neck (#978) * feat: add reduction for neck * feat: add reduction for neck * feat: add reduction for neck * feat:add linear reduction neck * feat: add reduction neck * mod out of linearReduction as tuple * fix typo * fix unit tests * fix unit tests Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> 2022-11-04 15:44:37 +08:00			`__all__ = [`
			`'GlobalAveragePooling', 'GeneralizedMeanPooling', 'HRFuseScales',`
			`'LinearReduction'`
			`]`