mmclassification/mmcls/models
Ma Zerun 9038c1c255
[Feature] Support TTA and add `--tta` in `tools/test.py`. (#1161)
* [Feature] Support TTA and add `--tta` in `tools/test.py`.

* Add unit tests.

* Rename the TTA model to `AverageClsScoreTTA`.
2022-12-30 11:46:17 +08:00
..
backbones [Feature] Add clip backbone. (#1258) 2022-12-22 16:33:57 +08:00
classifiers [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
heads [Feature] Support Multi-task. (#1229) 2022-12-30 10:36:00 +08:00
losses [Fix] Fix `loss_weight` in `LabelSmoothLoss`. (#1058) 2022-09-28 08:18:12 +08:00
necks [Feature] Add reduction for neck (#978) 2022-11-04 15:44:37 +08:00
retrievers [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
tta [Feature] Support TTA and add `--tta` in `tools/test.py`. (#1161) 2022-12-30 11:46:17 +08:00
utils [Feature] Support Multi-task. (#1229) 2022-12-30 10:36:00 +08:00
__init__.py [Feature] Support TTA and add `--tta` in `tools/test.py`. (#1161) 2022-12-30 11:46:17 +08:00
builder.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00