mmpretrain/mmcls/models/heads
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
..
__init__.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
base_head.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
cls_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
linear_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
multi_label_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
multi_label_linear_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
stacked_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00
vision_transformer_head.py [Refactor] Use post_process function to handle pred result processing. (#390) 2021-08-12 11:54:24 +08:00