fastreid.modeling.meta_arch.build Namespace Reference

Functions
	build_model (cfg, device=None)

Variables
	META_ARCH_REGISTRY = Registry("META_ARCH")

Detailed Description

@author:  liaoxingyu
@contact: sherlockliao01@gmail.com

Function Documentation

build_model()

fastreid.modeling.meta_arch.build.build_model	(		cfg,
			device = None )

Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``.
Note that it does not load any weights from ``cfg``.

Definition at line 18 of file build.py.

def build_model(cfg, device=None):
    """
    Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``.
    Note that it does not load any weights from ``cfg``.
    """
    meta_arch = cfg.MODEL.META_ARCHITECTURE
    model = META_ARCH_REGISTRY.get(meta_arch)(cfg)
    model.to(torch.device(cfg.MODEL.DEVICE))
    return model

Variable Documentation

META_ARCH_REGISTRY

fastreid.modeling.meta_arch.build.META_ARCH_REGISTRY = Registry("META_ARCH")

Definition at line 10 of file build.py.