mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
693596bc2f
* 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>
40 lines
671 B
Python
40 lines
671 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from mmcls.registry import MODELS
|
|
|
|
BACKBONES = MODELS
|
|
NECKS = MODELS
|
|
HEADS = MODELS
|
|
LOSSES = MODELS
|
|
CLASSIFIERS = MODELS
|
|
RETRIEVER = MODELS
|
|
|
|
|
|
def build_backbone(cfg):
|
|
"""Build backbone."""
|
|
return BACKBONES.build(cfg)
|
|
|
|
|
|
def build_neck(cfg):
|
|
"""Build neck."""
|
|
return NECKS.build(cfg)
|
|
|
|
|
|
def build_head(cfg):
|
|
"""Build head."""
|
|
return HEADS.build(cfg)
|
|
|
|
|
|
def build_loss(cfg):
|
|
"""Build loss."""
|
|
return LOSSES.build(cfg)
|
|
|
|
|
|
def build_classifier(cfg):
|
|
"""Build classifier."""
|
|
return CLASSIFIERS.build(cfg)
|
|
|
|
|
|
def build_retriever(cfg):
|
|
"""Build retriever."""
|
|
return RETRIEVER.build(cfg)
|