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

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

from .build import REID_HEADS_REGISTRY, build_reid_heads

# import all the meta_arch, so they will be registered
from .linear_head import LinearHead
from .bnneck_head import BNneckHead
from .arcface import ArcfaceHead
Update sampler code 2020-02-10 07:38:56 +08:00			`# encoding: utf-8`
			`"""`
			`@author: liaoxingyu`
			`@contact: sherlockliao01@gmail.com`
			`"""`

			`from .build import REID_HEADS_REGISTRY, build_reid_heads`

			`# import all the meta_arch, so they will be registered`
update version0.2 code 2020-03-25 10:58:26 +08:00			`from .linear_head import LinearHead`
			`from .bnneck_head import BNneckHead`
			`from .arcface import ArcfaceHead`