mmpretrain/mmcls/models
Ma Zerun f8f1700860
[Refactor] Use post_process function to handle pred result processing. (#390)
Use post_process function to handle pred result processing in `simple_test`.
2021-08-12 11:54:24 +08:00
..
backbones [Feature] Add transformer in transformer (#339) 2021-07-30 09:16:27 +08:00
classifiers [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
heads [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
losses [Fix] Use zero as default value of `thrs` in metrics. (#341) 2021-07-18 16:57:21 +08:00
necks GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test (#236) 2021-05-10 15:00:50 +08:00
utils [Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375) 2021-07-27 17:59:05 +08:00
__init__.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
builder.py [Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375) 2021-07-27 17:59:05 +08:00