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

12 lines
261 B
Python
Raw Normal View History

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
from .bn_linear import BNneckLinear
from .arcface import ArcFace