fastreid.modeling.heads.build Namespace Reference

Functions
	build_heads (cfg)

Variables
	REID_HEADS_REGISTRY = Registry("HEADS")

Detailed Description

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

Function Documentation

build_heads()

fastreid.modeling.heads.build.build_heads

(

cfg

)

Build REIDHeads defined by `cfg.MODEL.REID_HEADS.NAME`.

Definition at line 19 of file build.py.

def build_heads(cfg):
    """
    Build REIDHeads defined by `cfg.MODEL.REID_HEADS.NAME`.
    """
    head = cfg.MODEL.HEADS.NAME
    return REID_HEADS_REGISTRY.get(head)(cfg)

Variable Documentation

REID_HEADS_REGISTRY

fastreid.modeling.heads.build.REID_HEADS_REGISTRY = Registry("HEADS")

Definition at line 9 of file build.py.