7 Commits

Author SHA1 Message Date
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
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
mzr1996
ee6e585e41 Mazerun/base classifier 2022-07-18 11:11:13 +08:00
Ma Zerun
f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
yanglei
fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
yanglei
3a65f4529e Add Base and ResNet backbone 2020-06-12 14:40:27 +08:00
chenkai
a99e6eb10d init commit 2020-05-21 21:21:43 +08:00