fast-reid/fastreid/modeling/heads/__init__.py

# encoding: utf-8
"""
@author:  liaoxingyu
@contact: sherlockliao01@gmail.com
"""

from .build import REID_HEADS_REGISTRY, build_heads

# import all the meta_arch, so they will be registered
from .embedding_head import EmbeddingHead
from .clas_head import ClasHead
Update sampler code 2020-02-10 07:38:56 +08:00			`# encoding: utf-8`
			`"""`
			`@author: liaoxingyu`
			`@contact: sherlockliao01@gmail.com`
			`"""`

fix bug in mgn #272 Summary: fix get_norm bug in mgn 2020-09-17 18:17:53 +08:00			`from .build import REID_HEADS_REGISTRY, build_heads`
Update sampler code 2020-02-10 07:38:56 +08:00
			`# import all the meta_arch, so they will be registered`
refactor reid head Summary: merge BNneckHead, LinearHead and ReductionHead into EmbeddingHead because they are highly similar and can be prepared for ClsHead 2020-09-10 10:57:37 +08:00			`from .embedding_head import EmbeddingHead`
support classification in fastreid (#443) Summary: support classification and refactor build_dataloader which can support explicit parameters passing 2021-03-26 20:17:39 +08:00			`from .clas_head import ClasHead`