mmclassification/mmcls/engine/hooks
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
..
__init__.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00
class_num_check_hook.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
precise_bn_hook.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
retriever_hooks.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00
visualization_hook.py [Fix] Fix visualization hook. 2022-10-08 11:29:18 +08:00