mmclassification/tests/test_models
zzc98 693596bc2f
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055)
* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* update retriever

* fix lint

* add hook unit test

* Use `register_buffer` to save prototype vectors and add a progress bar
during preparing prototype.

* update UTs

* update UTs

* fix typo

* modify the hook

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-02 17:43:56 +08:00
..
test_backbones [Feature] Support MobileViT backbone. (#1068) 2022-10-18 17:05:59 +08:00
test_utils [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
test_classifiers.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
test_heads.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
test_losses.py [Improve] Update benchmark scripts (#1028) 2022-09-20 15:50:21 +08:00
test_necks.py
test_retrievers.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00